一个小时两道题,感觉难度也没有传说中的那么大,第二题稍微卡了一会,但是还是做完了
第一题
题意
有一个长度为n的全是小写字母的字符串,有m个要匹配的字符串,问一共能匹配多少次。
例如ababa,2个匹配字符串ab和aba,共能匹配4次。ab2次,aba两次。
数据范围n和m为1e5,匹配字符串长度为[2,10]。
分析
因为匹配字符串长度很短,因此完全可以计算所有的长度为[2,10]的子串的个数,然后用map来存,然后将m个匹配字符串的个数相加即可。
参考代码
#include <cstdio>
#include <iostream>
#include <string>
#include <unordered_map>
using namespace std;
int n, m;
unordered_map<string, int> M;
string s, t;
int main() {
cin >> n >> m;
cin >> s;
int len = s.length();
for (int i=0; i<len; ++i) {
for (int j=2; j<=10 && i+j-1<len; ++j) {
M[s.substr(i, j)]++;
}
}
int ans = 0;
for (int i=0; i<m; ++i) {
cin >> t;
ans += M[t];
}
cout << ans << endl;
return 0;
}
第二题
题意
给一个矩阵,其中值为-1的部分需要替换,如果为-1的部分上下左右相连可以组成连通块,每个连通块中的值都应该被替换成这与这个连通块直接相连的非-1部分的平均值向下取整。
例如:
5 4
0 8 -1 0
0 0 8 0
0 8 8 0
0 8 0 -1
0 -1 -1 0
输出:
0 8 5 0
0 0 8 0
0 8 8 0
0 8 0 0
0 2 2 0
矩阵中有三个连通块:
第一个:{(1,3)},与其直接相连的数有{8, 8, 0},求平均向下取整为5
第二个:{(4,4)},与其直接相连的数有{0, 0, 0},求平均向下取整为0
第三个:{(5,2), {5,3)},与其直接相连的数有{0, 8, 8, 0},求平均向下取整为2
分析
第一步肯定是先求连通块,然后给连通块编号。然后遍历每一个非连通块的值,判断其是否属于某一个连通块,并用sum和num计算每个连通块周围非-1的总和,以及个数。
参考代码
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
int n, m;
int a[1005][1005];
int id[1005][1005];
int vis[1005][1005];
int num[1000005];
int sum[1000005];
int dx[4] = { 0, 1, 0, -1};
int dy[4] = { 1, 0, -1, 0};
bool judge(int x, int y) {
if (1<=x && x<=n && 1<=y && y<=m) return true;
return false;
}
void dfs(int x, int y, int _id) {
vis[x][y] = 1;
id[x][y] = _id;
for (int i=0; i<4; ++i) {
int nx = x + dx[i];
int ny = y + dy[i];
if (judge(nx, ny) && !vis[nx][ny] && a[nx][ny]==-1) {
dfs(nx, ny, _id);
}
}
}
int main() {
scanf("%d%d", &n, &m);
memset(sum, 0, sizeof(sum));
memset(num, 0, sizeof(num));
for (int i=1; i<=n; ++i) {
for (int j=1; j<=m; ++j) {
id[i][j] = vis[i][j] = 0;
scanf("%d", &a[i][j]);
}
}
// 求连通块,并编号
int __id = 0;
for (int i=1; i<=n; ++i) {
for (int j=1; j<=m; ++j) {
if (a[i][j]==-1 && !vis[i][j]) {
__id++;
dfs(i, j, __id);
}
}
}
// 计算连通块周围非-1的数量以及总和
for (int i=1; i<=n; ++i) {
for (int j=1; j<=m; ++j) {
if (a[i][j] != -1) {
int idd[4];
idd[0] = idd[1] = idd[2] = idd[3] = -1;
// 判断是否属于上下左右的连通块,如果属于则累加和以及个数
// 注意有可能出现上下左右其中几个是同一个连通块,需要去重
for (int k=0; k<4; ++k) {
int nx = i + dx[k];
int ny = j + dy[k];
if (judge(nx, ny) && a[nx][ny]==-1) {
bool flag = true;
for (int l=0; l<k; ++l) {
if (idd[l]!=-1 && idd[l]==id[nx][ny]) {
flag = false;
break;
}
}
if (!flag) continue;
idd[k] = id[nx][ny];
sum[id[nx][ny]] += a[i][j];
num[id[nx][ny]]++;
}
}
}
}
}
for (int i=1; i<=n; ++i) {
for (int j=1; j<=m; ++j) {
if (a[i][j] == -1) {
if (num[id[i][j]] == 0)
printf("0 ");
else
printf("%d ", sum[id[i][j]] / num[id[i][j]]);
} else {
printf("%d ", a[i][j]);
}
}
printf("\n");
}
return 0;
}