White Lines CodeForces - 1200D (前缀和，思维)

题目大意

给定一个n * n的红黑格子，然后你可以选一个点把 以这个点为左上角的k * k的区域都变为白的，问最多多少个一行或一列都为白。

思路

前缀和思路，预处理出每一行的前缀和cr[i][j]，然后对每一行来说，在算一个每个黑格的贡献，贡献就是如果涂白这行j到 j + k - 1的网格这行变为全白，那么这个格子的贡献就为1(c[i][j] == 1) ，每一列同样的预处理，之后求出每个c[i][j]对于行的前缀和，这样就可以求出i行到i+k - 1的区间和，对于列的操作同理。
然后对于每个点求出两个前缀和相加，选一个最大值即可。

代码

#include <map>
#include <set>
#include <cmath>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;

typedef long long ll;
int n,k;
const int maxn = 2000 + 20;
char s[maxn];
int cr[maxn][maxn], cc[maxn][maxn], r[maxn][maxn], c[maxn][maxn], sr[maxn][maxn], sc[maxn][maxn];
int main() {
   // freopen("/Users/maoxiangsun/MyRepertory/input.txt", "r", stdin);
    cin>>n>>k;
    for(int i = 1; i <= n; i++) {
        scanf("%s", s + 1);
        for(int j = 1; j <= n; j++) {
            cr[i][j] = cr[i-1][j] + (s[j] == 'B');
            cc[i][j] = cc[i][j-1] + (s[j] == 'B');
        }
    }
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n - k + 1; j++) {
            if(cc[i][n] && cc[i][j + k - 1] - cc[i][j - 1] == cc[i][n]) c[i][j] = 1;
        }
    }
    for(int j = 1; j <= n; j++) {
        for(int i = 1; i <= n - k + 1; i++) {
            if(cr[n][j] && cr[i + k - 1][j] - cr[i - 1][j] == cr[n][j]) r[i][j] = 1;
        }
    }
    int cnt = 0;
    for(int i = 1; i <= n; i++) {
        cnt+=(cr[n][i] == 0);
        cnt+=(cc[i][n] == 0);
    }
    //cout << cnt<<endl;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n; j++) {
            sr[i][j] = sr[i][j-1] + r[i][j];
            sc[i][j] = sc[i-1][j] + c[i][j];
        }
    }
    int ans = 0;
    for(int i = 1; i <= n - k + 1; i++) {
        for(int j = 1; j <= n - k + 1; j++) {
            ans = max(ans, sr[i][j + k - 1] - sr[i][j - 1] + sc[i + k - 1][j] - sc[i - 1][j]);
        }
    }
    cout<<ans+cnt<<endl;
    return 0;
}
/*
 4 2
 BWWW
 WBBW
 WBBW
 WWWB
 */