NC53676 「土」秘法地震（二维前缀和）

链接：https://ac.nowcoder.com/acm/problem/53676

题目描述

帕秋莉掌握了一种土属性魔法
这种魔法可以在一片k×k大小的一个正方形区域内产生地震
但是如果某片即将产生地震的区域内有建筑物，帕秋莉会停止施法
整个地图大小为n×m，其中一些地方有建筑
请问有多少种可能的情况，使得帕秋莉会停止施法

输入描述:

第一行三个数n, m, k，意义见描述
接下来一个n×m的01矩阵表示这篇区域的情况，1表示这个地方有建筑

输出描述:

输出一个数表示答案

输入

4 4 2
1000
0100
0000
0001

输出

5

备注:

对于30%的数据，n, m≤30
对于100%的数据，n, m≤1000，k≤min(n, m)

---

solution

最直接的方法就是暴力，我看这个题直接想到的就是二维前缀和，$a[i][j]$ 是左上角$(0,0)$到右下角$(i,j)$这部分的和。计算时注意容斥，自己画一下很简单。

code

#include <bits/stdc++.h>
using namespace std;
int n, m, k;
int a[1010][1010];
int main() {
    ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
    cin >> n >> m >> k;
    for (int i = 1; i <= n; i++) {
        string s;
        cin >> s;
        for (int j = 1; j <= m; j++) {
            a[i][j] = s[j - 1] - '0' + a[i][j - 1] + a[i - 1][j] - a[i - 1][j - 1];
        }
    }
    int ans = 0;
    for (int i = 1; i <= n - k + 1; i++) {
        for (int j = 1; j <= m - k + 1; j++) {
            if (a[i + k - 1][j + k - 1] - a[i + k - 1][j - 1] - a[i - 1][j + k - 1] + a[i - 1][j - 1]) {
                ans++;
            }
        }
    }
    cout << ans << endl;
    return 0;
}