2020阿里笔试题解(9.11)

一个小时两道题，感觉难度也没有传说中的那么大，第二题稍微卡了一会，但是还是做完了

第一题

题意

有一个长度为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 40 8 -1 00 0 8 00 8 8 00 8 0 -10 -1 -1 0

输出：

0 8 5 00 0 8 00 8 8 00 8 0 00 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;}