牛客网 「土」秘法地震 (DP)

题目：

解题思路：
给出n，m，k 三个数。
输入n * m 的矩阵，矩阵元素为 0 或 1。
我们只需要计算有多少k * k 的元素不全为0
的矩阵即可。采用DP思想。

eg： 4 4 2

4 * 4 的方格，然后求出有多少 k * k 大小的矩阵中元素全部为 0。

首先输入每个单元格中的元素值，然后我们采用递归来计算当前位置的左上方全部元素的和。

for (int i = 1; i <= n; i++){
        for (int j = 1; j <= m ; j++){
            ans[i][j] = ans[i - 1][j] + ans[i][j - 1] - ans[i - 1][j - 1]+ ans[i][j];
        }
    }

比如i = 3，j = 3时：
ans[i ][ j - 1 ] ： 橙色区域

ans[ i - 1][ j ]: 红色区域

所以要减去多余的ans [ i - 1][j - 1] :绿色
加上少的 ans [ i ][ j ] ： 蓝色

最终得到当前的 ans[ i ] [ j ] （ 当前坐标左上角的元素和）

接下来根据范围大小遍历判定每个k * k 的矩阵是否符合要求：

int result = 0;
    for (int i = 1; i <= n - k + 1; i++){
        for (int j = 1; j <= m -k +1; j++){
            if (ans[i + k -1][j + k -1] - (ans[i - 1][j + k -1]+ans[i + k -1][j - 1]-ans[i - 1][j - 1]) != 0){
                result ++;
            }
        }
    }

道理和前面相同，手写下就可以得出。

完整代码：

#include <iostream>
#include <cstdio>

using namespace std;

int ans[1005][1005];

int main(){
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    char c;
    for (int i = 1; i <= n;i ++){
        for (int j = 1; j <= m; j ++){
            cin>>c;
            ans[i][j] = c - '0';
        }
    }
    for (int i = 1; i <= n; i++){
        for (int j = 1; j <= m ; j++){
            ans[i][j] = ans[i - 1][j] + ans[i][j - 1] - ans[i - 1][j - 1]+ ans[i][j];
        }
    }
    int result = 0;
    for (int i = 1; i <= n - k + 1; i++){
        for (int j = 1; j <= m -k +1; j++){
            if (ans[i + k -1][j + k -1] - (ans[i - 1][j + k -1]+ans[i + k -1][j - 1]-ans[i - 1][j - 1]) != 0){
                result ++;
            }
        }
    }
    printf("%d\n",result);
    return 0;

}