牛客-模拟、枚举与贪心 2022.11.11

「土」秘法地震

二维前缀和，如果k*k区域的前缀和为0，说明当前区域一个建筑都没有，不会影响，否则会影响。

#include <bits/stdc++.h>
using namespace std;
int num[2010][2010];
int main() {
    memset(num, 0, sizeof(num));
    int n, m, k;
    cin >> n >> m >> k;
    for(int i = 1; i <= n; i++) {
        string s;
        cin >> s;
        for(int j = 1; j <= m; j++) {
            num[i][j] = num[i - 1][j] + num[i][j - 1] - num[i - 1][j - 1];
            num[i][j] += (s[j - 1] - '0'); 
        }
    }
    int res = 0;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= m; j++) {
            if(i - k >= 0 && j - k >= 0) {
                int tmp = num[i][j] - num[i - k][j] - num[i][j - k] + num[i - k][j - k];
                if(tmp != 0)
                    res += 1;
            }
        }
    }
    cout << res << endl;
    return 0;
}

仓库选址

记录所有有货物xu qi