poj 1681 Painter's Problem

高斯消元，套模板

#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <cstdio>
#include <string>
#include <cstring>
using namespace std;
#define CL(a,b) memset(a,b,sizeof(a))
#define INF 100000000
const int M(300);
const int N(20);
int a[M][M];
int free_x[M],num;
int x[M];
char str[N][N];
void init(int n)
{
	int i,j,x;
	CL(a,0);
	for(i=0;i<n;i++)
		for(j=0;j<n;j++)
		{
			x=i*n+j;
			a[x][x]=1;
			if(i-1>=0)
				a[x][n*(i-1)+j]=1;
			if(i+1<n)
				a[x][n*(i+1)+j]=1;
			if(j-1>=0)
				a[x][n*i+j-1]=1;
			if(j+1<n)
				a[x][n*i+j+1]=1;
		}
}
int gauss(int equ,int var)
{
	int i,j,k,col;
	int max_r;
	num=0;
	CL(free_x,0);
	CL(x,0);
	for(k=0,col=0;k<equ&&col<var;k++,col++)
	{
		max_r=k;
		for(i=k+1;i<equ;i++)
			if(a[i][col]>a[max_r][col])
				max_r=i;
		if(a[max_r][col]==0)
		{
			k--;
			free_x[num++]=col;
			continue;
		}
		if(k!=max_r)
			for(i=0;i<var+1;i++)
				swap(a[k][i],a[max_r][i]);
		for(i=0;i<equ;i++)
			if(i!=k&&a[i][col]!=0)
				for(j=col;j<var+1;j++)
					a[i][j]=a[i][j]^a[k][j];
		
	
	}
/*	printf("\n");
	for(i=0;i<equ;i++)
		{
			for(j=0;j<var+1;j++)
				printf("%d ",a[i][j]);
			printf("\n");
		}*/
	for(i=k;i<equ;i++)
	{
		if(a[i][col]!=0)
			return -1;
	}
//	printf("%d\n",num);
	int s=(1<<(var-k)),index,cnt=0,l,ret=INF;
	//if(num!=var-k)
	//	printf("error\n");
	for(i=0;i<=0;i++)
	{
		index=i;cnt=0;
		CL(x,0);
		for(j=0;j<var-k;j++)
		{
			x[free_x[j]]=(index&1);
			if(x[free_x[j]])
				cnt++;
			index>>=1;
		}
		for(j=k-1;j>=0;j--)
		{
			int tmp=a[j][var];
			for(l=j+1;l<var;l++)
			{
				if(a[j][l])
					tmp ^= x[l] ;
			}
			x[j] = tmp ;
			if(x[j])
				cnt++;
		}
	//	printf("%d\n",cnt);
		if(ret>cnt)
			ret=cnt;
	}
	return ret;
}
int main()
{
	int t,n,i,j;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		init(n);
		for(i=0;i<n;i++)
		{
			scanf("%s",str[i]);
			for(j=0;j<n;j++)
				if(str[i][j]=='w')
					a[n*i+j][n*n]=1;
		}
	/*	for(i=0;i<n*n;i++)
		{
			for(j=0;j<n*n+1;j++)
				printf("%d ",a[i][j]);
			printf("\n");
		}*/
		int k=gauss(n*n,n*n);
		if(k==-1)
			printf("inf\n");
		else
			printf("%d\n",k);
	}
}