Codeforces 1200D. White Line(思维 + 前缀和)

戳我看题

题意： 给一个字符矩阵，只含有 W 和 B，现在有一种操作，将一个 k * k 矩阵全部变为 W，问最多能有多少个行和列全部为 W

思路： 可以单独算每一个点的贡献。
以行为例：

• 先算的每一行的 B 的前缀和，枚举这一行的所有点，若 $prex[j+k-1]-prex[j-1]==prex[n]\&\&prex[n]!=0$，那么 $j$ 这个点的贡献值就是 1，最后对 $j$ 这个贡献值 在列的方向上做前缀和。
• 列也是如此，先对列做前缀后，然后算贡献值，在 行的方向上对贡献值做前缀和
• 最后枚举每一个点作为操作的左上角，利用两个贡献值的前缀和可以 $O(1)$ 的求得这个大小为 k * k 的矩阵增加了多少个满足条件的行列，去 max
• 加上本来就满足条件的行列，即 ans 。。。。不懂的 最好看代码，可能会更加明白一些

Code：

#include<bits/stdc++.h>
#define debug(x) cerr << "[" << #x <<": " << (x) <<"]"<< endl
#define pii pair<int,int>
#define clr(a,b) memset((a),b,sizeof(a))
#define rep(i,a,b) for(int i = a;i < b;i ++)
#define pb push_back
#define MP make_pair
#define LL long long
#define ull unsigned LL
#define ls i << 1
#define rs (i << 1) + 1
#define fi first
#define se second
#define ptch putchar
#define CLR(a) while(!(a).empty()) a.pop()

using namespace std;
inline LL read() {
    LL s = 0,w = 1;
    char ch = getchar();
    while(!isdigit(ch)) {
        if(ch == '-') w = -1;
        ch = getchar();
    }
    while(isdigit(ch))
        s = s * 10 + ch - '0',ch = getchar();
    return s * w;
}
inline void write(LL x) {
    if(x < 0)
        putchar('-'), x = -x;
    if(x > 9)
        write(x / 10);
    putchar(x % 10 + '0');
}
#ifndef ONLINE_JUDGE
    clock_t prostart = clock();
#endif

const int maxn = 2010;
char mp[maxn][maxn];
int prex[maxn],prey[maxn];
int sumx[maxn][maxn],sumy[maxn][maxn];

int main() {
#ifndef ONLINE_JUDGE
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
#endif

    int n;
    while(~scanf("%d",&n)){
        int k = read();
        for(int i = 1;i <= n;++ i)
            scanf("%s",mp[i] + 1);
        int tmpx = 0,tmpy = 0;
        for(int i = 1;i <= n;++ i){
            for(int j = 1;j <= n;++ j){
                prex[j] = prex[j - 1] + (mp[i][j] == 'B');
            }
            if(prex[n] == 0) ++ tmpx;
            for(int j = 1;j + k - 1 <= n;++ j){
                sumx[i][j] = sumx[i - 1][j] + (prex[j + k - 1] - prex[j - 1] == prex[n] && prex[n] != 0);
            }
        }

        for(int j = 1;j <= n;++ j){
            for(int i = 1;i <= n;++ i){
                prey[i] = prey[i - 1] + (mp[i][j] == 'B');
            }
            if(prey[n] == 0) ++ tmpy;
            for(int i = 1;i + k - 1 <= n;++ i){
                sumy[i][j] = sumy[i][j - 1] + (prey[i + k - 1] - prey[i - 1] == prey[n] && prey[n] != 0);
            }
        }

        int ans = 0;
        for(int i = 1;i + k - 1 <= n;++ i)
            for(int j = 1;j + k - 1 <= n;++ j){
                ans = max(ans,sumx[i + k - 1][j] - sumx[i - 1][j] + sumy[i][j + k - 1] - sumy[i][j - 1]);
            }
        write(ans + tmpx + tmpy); ptch('\n');
    }

#ifndef ONLINE_JUDGE
    cout << "time: " << 1.0 * (clock() - prostart) / CLOCKS_PER_SEC << " s" << endl;
#endif
    return 0;
}