Codeforces Round #578 (Div. 2) D. White Lines（二维差分、二维前缀和）（详细）

题目链接

input

4 2
BWWW
WBBW
WBBW
WWWB

output

4

input

3 1
BWB
WWB
BWB

output

2

input

5 3
BWBBB
BWBBB
BBBBB
BBBBB
WBBBW

output

2

大致题意：

给定一个n*n的矩阵，矩阵元素非'W'即'B'，一次操作将k*k的元素全都变为'B'，问一次操作后White Lines即一行或一列全都是'W'的Line最多有多少。

 

题解：

以sum[i][j]表示该次操作的左上角坐标为（i，j）时白线的最大数目。行和列是一样的，下面考虑行：

• 若一行全是'B'，无论操作的左上角坐标在哪，这一行都会对最终的答案贡献1；
• 若一行不全是'B'，求出最左边'B'的坐标L，最右边'B'的坐标R，若R-L+1>k即无论怎么操作都不能将这一行全部变为'W'，即不会对答案产生贡献；
• 反之，如果R-L+1<=k意味着能将这一行变为全'W'，不过只有操作的左上角坐标满足要求时才能对答案产生贡献。见下图：

 

 

可以看出，操作的左上角坐标的范围是[max(1,i-k+1)，max(1,R-k+1)]为左上角，[i，L]为右下角的矩形区域中。

也就是说，上述情况只对sum[x][y]，（ max(1,i-k+1) <= x<= i ，max(1,R-k+1) <= y <= L）产生贡献1。

 

列 同理。最后取最大的sum[i][j]即可。接下来是怎么高效地实现这个过程：

前置知识：二维差分、二维前缀和

AC代码：
 

#include<iostream>
#include<sstream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<iomanip>
#include<cstdlib>
#include<cctype>
#include<vector>
#include<string>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<map>
#include<set>
#define mem(a,b) memset(a,b,sizeof(a))
#define random(a,b) (rand()%(b-a+1)+a)
#define ll long long
#define ull unsigned long long
#define e 2.71828182
#define Pi acos(-1.0)
#define ls(rt) (rt<<1)
#define rs(rt) (rt<<1|1)
#define lowbit(x) (x&(-x))
using namespace std;
const int MAXN=2e3+5;
int sum[MAXN][MAXN],n,k;
char paint[MAXN][MAXN];
int add[MAXN][MAXN];
int read()
{
	int s=1,x=0;
	char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') s=-1;ch=getchar();}
	while(isdigit(ch)) {x=10*x+ch-'0';ch=getchar();}
	return x*s;
}
int main()
{
	n=read(),k=read();
	for(int i=1;i<=n;++i)
	cin>>paint[i]+1;

	for(int i=1;i<=n;++i)//处理每一行对sum[x][y]的贡献 
	{
		int L=0,R=0;//每一行最左的B和最右的B的下标 
		for(int j=1;j<=n;++j)
		{
			if(paint[i][j]=='B')
			{
				if(L==0) L=j,R=j;
				else R=j;
			}
		}
		if(L==0) {add[1][1]++;continue;}//这一行全白，+1
		if(R-L+1>k) continue;//这一行怎么擦也不能全白
		//二维差分相关操作 ，前缀和修改操作 
		int x1=max(1,i-k+1),y1=max(1,R-k+1),x2=i,y2=L;
		add[x1][y1]++,add[x2+1][y1]--,add[x1][y2+1]--,add[x2+1][y2+1]++; 
	}
	
	for(int j=1;j<=n;++j)
	{
		int U=0,D=0;//每一列最上的B和最下的B的下标 
		for(int i=1;i<=n;++i)
		{
			if(paint[i][j]=='B')
			{
				if(U==0) U=i,D=i;
				else D=i;
			}
		}
		if(D==0) {add[1][1]++;continue;}//这一列全白，+1
		if(D-U+1>k) continue; //这一列怎么擦也不能全白
		//
		int x1=max(1,D-k+1),y1=max(1,j-k+1),x2=U,y2=j;
		add[x1][y1]++,add[x2+1][y1]--,add[x1][y2+1]--,add[x2+1][y2+1]++;
	}
	
	int ans=0;
	for(int i=1;i<=n;++i)
	for(int j=1;j<=n;++j)
	{
		sum[i][j]=sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1]+add[i][j];//计算二维前缀和 
		ans=max(ans,sum[i][j]);
	}
	cout<<ans<<endl;
}

 

也可以只用一个sum数组：

#include<iostream>
#include<sstream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<iomanip>
#include<cstdlib>
#include<cctype>
#include<vector>
#include<string>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<map>
#include<set>
#define mem(a,b) memset(a,b,sizeof(a))
#define random(a,b) (rand()%(b-a+1)+a)
#define ll long long
#define ull unsigned long long
#define e 2.71828182
#define Pi acos(-1.0)
#define ls(rt) (rt<<1)
#define rs(rt) (rt<<1|1)
#define lowbit(x) (x&(-x))
using namespace std;
const int MAXN=2e3+5;
int sum[MAXN][MAXN],n,k;
char paint[MAXN][MAXN];
int read()
{
	int s=1,x=0;
	char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') s=-1;ch=getchar();}
	while(isdigit(ch)) {x=10*x+ch-'0';ch=getchar();}
	return x*s;
}
int main()
{
	n=read(),k=read();
	for(int i=1;i<=n;++i)
	cin>>paint[i]+1;

	for(int i=1;i<=n;++i)//处理每一行对sum[x][y]的贡献 
	{
		int L=0,R=0;//每一行最左的B和最右的B的下标 
		for(int j=1;j<=n;++j)
		{
			if(paint[i][j]=='B')
			{
				if(L==0) L=j,R=j;
				else R=j;
			}
		}
		if(L==0) {sum[1][1]++;continue;}//这一行全白，+1
		if(R-L+1>k) continue;//这一行怎么擦也不能全白
		//二维差分相关操作 ，前缀和修改操作 
		int x1=max(1,i-k+1),y1=max(1,R-k+1),x2=i,y2=L;
		sum[x1][y1]++,sum[x2+1][y1]--,sum[x1][y2+1]--,sum[x2+1][y2+1]++; 
	}
	
	for(int j=1;j<=n;++j)
	{
		int U=0,D=0;//每一列最上的B和最下的B的下标 
		for(int i=1;i<=n;++i)
		{
			if(paint[i][j]=='B')
			{
				if(U==0) U=i,D=i;
				else D=i;
			}
		}
		if(D==0) {sum[1][1]++;continue;}//这一列全白，+1
		if(D-U+1>k) continue; //这一列怎么擦也不能全白
		//
		int x1=max(1,D-k+1),y1=max(1,j-k+1),x2=U,y2=j;
		sum[x1][y1]++,sum[x2+1][y1]--,sum[x1][y2+1]--,sum[x2+1][y2+1]++;
	}
	
	int ans=0;
	for(int i=1;i<=n;++i)
	for(int j=1;j<=n;++j)
	{
		sum[i][j]+=sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1];//计算二维前缀和 
		ans=max(ans,sum[i][j]);
	}
	cout<<ans<<endl;
}