D. White Lines

D. White Lines

给定一个$n\times n$的$WB$矩阵，给定一个$k*k$的能把$B$变成$W$的橡皮擦，求橡皮擦作用一次后，全为$W$的行、列总数最大值

前缀和差分

#include<bits/stdc++.h>
using namespace std;
int A[][];
char C[][];
int B[][];
int E[][];
int  F[][];
int n,k;
int main()
{
    scanf("%d%d",&n,&k);
    int ans=;
    for(int i=; i<=n; i++)
    {
        getchar();
        for(int j=; j<=n; j++)
        {
            C[i][j]=getchar();
            A[i][j]=A[i][j-]+(C[i][j]=='B');
        }
        if(A[i][n]==)ans++;
    }
    for(int i=;i<=n;i++){
        for(int j=;j<=n;j++){
            B[j][i]=B[j-][i]+(C[j][i]=='B');
        }
        if(B[n][i]==)ans++;
    }
    //cout<<ans<<'\n';
    int t=,ma=;
    for(int i=;i<=n;i++){
        for(int j=;j+k-<=n;j++){
            E[i][j]=E[i-][j]+(A[i][j+k-]-A[i][j-]==A[i][n]&&A[i][n]);
        }
    }
    for(int i=;i<=n;i++){
        for(int j=;j+k-<=n;j++){
            F[j][i]=F[j][i-]+(B[j+k-][i]-B[j-][i]==B[n][i]&&B[n][i]);
        }
    }
    for(int i=;i+k-<=n;i++){
        for(int j=;j+k-<=n;j++){
            t=E[i+k-][j]-E[i-][j];
            t+=F[i][j+k-]-F[i][j-];
            ma=max(t,ma);
        }
 
    }
    cout<<ans+ma<<'\n';
 
}