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;
}