「土」秘法地震(二维前缀和)

「土」秘法地震

https://ac.nowcoder.com/acm/problem/53676


题目:

给一个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;
}
全部评论

相关推荐

昨天 00:11
已编辑
广东工业大学 算法工程师
避雷深圳&nbsp;&nbsp;yidao,试用期&nbsp;6&nbsp;个月。好嘛,试用期还没结束,就直接告诉你尽快找下一家吧,我谢谢您嘞
牛客75408465号:笑死,直属领导和 hr 口径都没统一,各自说了一些离谱的被裁理由,你们能不能认真一点呀,哈哈哈哈哈😅😅😅
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务