题目:
给一个nm的01矩阵。问有多少个kk的含1矩阵。(n最大1000)
做法:
二维前缀和。跟板子没啥区别。。
利用二维前缀和可以快速得到某矩形区域中数字的和。那么枚举所有k*k矩阵的右下角位置,利用二维前缀和得到矩阵中1的数量,大于零则ans++即可。
代码:
#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false), cin.tie(0)
#define debug(a) cout << #a ": " << a << endl
using namespace std;
typedef long long ll;
const int N = 1010;
char s[N];
int prefix[N][N];
int main(void){
IOS;
int n, m, k; cin >> n >> m >> k;
for (int i = 1; i <= n; ++i){
cin >> (s+1);
for (int j = 1; j <= m; ++j){
if (s[j] == '1'){
prefix[i][j] = 1;
}
}
}
for (int i = 1; i <= n; ++i){
for (int j = 1; j <= m; ++j){
prefix[i][j] += prefix[i-1][j]+prefix[i][j-1]-prefix[i-1][j-1];
}
}
int ans = 0;
for (int i = k; i <= n; ++i){
for (int j = k; j <= m; ++j){
int cnt = prefix[i][j]-prefix[i][j-k]-prefix[i-k][j]+prefix[i-k][j-k];
if (cnt) ans++;
}
}
cout << ans << endl;
return 0;
}
京公网安备 11010502036488号