Rectangle Filling 题解

题目大意

每次选择的位置 $(x1,y1)$ 和 $(x2,y2)$，它们目前都是相同的颜色 $c$，则将 $\min (x1,x2)\le x\le \max (x1,x2)$ 和 $\min (y1,y2)\le y\le \max (y1,y2)$ 中所有 $(x,y)$ 的颜色设置为 $c$。做以上操作知道（分辨出是）无解或所有颜色统一。

分析

根据样例可以发现：只要有（至少一组）对顶顶点的颜色可以相同，则有解，否则无解。那么只需要判断四个点的颜色与其能否变色即可。时间复杂度 $O(nm+n+m)$。

代码如下

#include<bits/stdc++.h>
using namespace std;
const int N=1000;
int n,m,g[N][N],bz[N];
signed main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		bz[1]=bz[2]=bz[3]=bz[4]=0;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)
		{
			int j=0;
			char ch=getchar();
			while(ch!='W'&&ch!='B') ch=getchar();
			while(ch=='W'||ch=='B')
			{
				j++;
				if(ch=='B') g[i][j]=1;
				else g[i][j]=0;
				ch=getchar();
			}
		}
		if(g[n][1]==g[1][m]||g[1][1]==g[n][m])
		{
			printf("YES\n");
		}
		else
		{
			int ff=0;
			for(int i=1;i<=n;i++)
			{
				if(g[i][1]!=g[1][1]&&bz[1]==0)
				{
					bz[1]++;
				}
				if(g[i][m]!=g[1][m]&&bz[2]==0)
				{
					bz[2]++;
				}
				if(g[i][1]!=g[n][1]&&bz[3]==0)
				{
					bz[3]++;
				}
				if(g[i][m]!=g[n][m]&&bz[4]==0)
				{
					bz[4]++;
				}
			}
			for(int i=1;i<=m;i++)
			{
				if(g[1][i]!=g[1][1]&&bz[1]==1)
				{
					
					ff=1;
					printf("YES\n");
					break;
				}
				if(g[1][i]!=g[1][m]&&bz[2]==1)
				{
					ff=1;
					printf("YES\n");
					break;
				}
				if(g[n][i]!=g[n][1]&&bz[3]==1)
				{
					ff=1;
					printf("YES\n");
					break;
				}
				if(g[n][i]!=g[n][m]&&bz[4]==1)
				{
					ff=1;
					printf("YES\n");
					break;
				}
			}
			if(ff==0) printf("NO\n");
		}
	}
	return 0;
}

到此结束，谢谢！