51nod 1448 二染色问题

开始还以为要用二维线段树优化，然后暴力水过了。。。

时间复杂度O(n^6)

#include<bits/stdc++.h>
using namespace std;

char mz[22][22];
int a[22][22];
int n,k;

int cal(int x1,int y1,int x2,int y2)
{
	int i,j,ret=0;
	for(i=x1;i<=x2;i++)
		for(j=y1;j<=y2;j++)
			ret+=a[i][j];
	return ret;
}

int main()
{
	int T,i,j,x,y,flag,tmp;
	cin>>T;
	while(T--)
	{
		scanf("%d%d",&n,&k);
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
				scanf(" %c",&mz[i][j]);
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
			{
				if(mz[i][j]=='B')
					a[i][j]=10000;
				else
					a[i][j]=1;
			}
		} 
		for(;;)
		{
			flag=0;
			for(i=1;i<=n-k+1;i++)
			{
				for(j=1;j<=n-k+1;j++)
				{
					tmp=cal(i,j,i+k-1,j+k-1);
					if(tmp==0)
						continue;
					if(tmp/10000&&tmp%10000)
						continue;
					for(x=i;x<=i+k-1;x++)
						for(y=j;y<=j+k-1;y++)
							a[x][y]=0;
					flag=1;
				}
			} 
			if(!flag)
				break;
		}
		if(cal(1,1,n,n))
			printf("Impossible\n");
		else
			printf("Possible\n");
	}
}