【Codeforce】 Round #578 (Div. 2) D.White Lines（前缀和统计处理）

题目链接

题目大意：现在给你N*N的矩阵，矩阵的每一个格子都有一个颜色，现在我们可以进行一次涂色的操作，这个操作只能进行一次，能够把K*K格子变成白色，然后问你最多能够有多少个白色的直线，

一行都是白的是一条直线，一列都是白色的同理

思路：这个题目主要是因为他是行和列的状态都是不冲突的，所以我是每一种都搞了两个数组来搞。

我是这样搞的，使用了两个前缀和，计算出某一行或者是某一列的前缀，然后判断在这一个点上染色的话是不是能够把某一行或者是列通上直线，如果可以的话把一个数组fl或者是fh置1，表示在行或者是列上这一个点是可行的点，

最后的话，就是把每一行或者是列的可行的点再做一遍前缀和，然后枚举每一个点，看看对这个点做K*K的染色后最多是多少

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
const int maxn=2e3+10;
char mp[maxn][maxn];
int sumh[maxn][maxn];//表示第i行 第0~j列的前缀和
int suml[maxn][maxn];//表示第i列 第0~j行的前缀和
int fh[maxn][maxn],fl[maxn][maxn];//表示行或者是列上可行的点的位置
int sx[maxn][maxn],sy[maxn][maxn];//表示第j列上 0~i行的前缀和  ++ 第j行上第i列的前缀和
int n,k;
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
            cin>>mp[i][j];
    int cnt=0;
    for(int i=1; i<=n; i++) ///hang
    {
        for(int j=1; j<=n; j++)
        {
            sumh[i][j]=sumh[i][j-1];
            if(mp[i][j]=='B')
                sumh[i][j]++;
        }
        if(sumh[i][n]==0)
            cnt++;
    }
    for(int i=1; i<=n; i++) ///lie
    {
        for(int j=1; j<=n; j++)
        {
            suml[i][j]=suml[i][j-1];
            if(mp[j][i]=='B')
                suml[i][j]++;
        }
        if(suml[i][n]==0)
            cnt++;
    }
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
        {
            if(sumh[i][n]!=0)  //判断可行的在第i行上的点
                if(sumh[i][j+k-1]-sumh[i][j-1]==sumh[i][n])
                    fh[i][j]=1;
            if(suml[i][n]!=0)  //判断可行的在第i列上的点
                if(suml[i][j+k-1]-suml[i][j-1]==suml[i][n])
                    fl[i][j]=1;
        }


   for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            sx[i][j]=sx[i-1][j]+fh[i][j];  //对前i行做前缀和，
            sy[i][j]=sy[i-1][j]+fl[i][j];
        }

    int ans=0;
    for(int i=1; i<=n-k+1; i++)
        for(int j=1; j<=n-k+1; j++)
        {  //求最大的（i+k-1~i）行第j列的可行的点的和，最大的（i+k-1~i）列第j行的可行的点的和
            ans=max(ans,sx[i+k-1][j]-sx[i-1][j]+sy[j+k-1][i]-sy[j-1][i]+cnt);
        }
    printf("%d\n",ans);
}