POJ 1681 Painter's Problem (高斯消元)

题目地址：POJ 1681

跟前两题几乎一模一样的。。。不多说了。高斯消元+自由元枚举。

代码如下：

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
using namespace std;
#define LL __int64
#define pi acos(-1.0)
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double eqs=1e-6;
char mp[20][20];
int a[300][300], free_x[300], free_num, state[300];
int jx[]={0,0,1,-1};
int jy[]={1,-1,0,0};
int gauss(int n)
{
int i, j, k, h, tmp, max_r;
free_num=0;
for(i=0,j=0;i<n&&j<n;i++,j++){
max_r=i;
for(k=i+1;k<n;k++){
if(a[k][j]>a[max_r][j]) max_r=k;
}
if(max_r!=i){
for(k=j;k<=n;k++){
swap(a[i][k],a[max_r][k]);
}
}
if(!a[i][j]){
free_x[free_num++]=j;
i--;
continue ;
}
for(k=i+1;k<n;k++){
if(a[k][j]){
for(h=j;h<=n;h++){
a[k][h]^=a[i][h];
}
}
}
}
//printf("%d\n",free_num);
tmp=i;
for(i=tmp;i<n;i++){
if(a[i]
){
return -1;
}
}
int tot=1<<free_num;
int ans=INF, cnt;
for(i=0;i<tot;i++){
cnt=0;
for(j=0;j<free_num;j++){
if(i&(1<<j)){
state[free_x[j]]=1;
cnt++;
}
else state[free_x[j]]=0;
}
for(j=tmp-1;j>=0;j--){
int t=0;
while(a[j][t]==0) t++;
state[t]=a[j]
;
for(k=t+1;k<n;k++){
if(a[j][k]) state[t]^=state[k];
}
cnt+=state[t];
}
//printf("%d\n",cnt);
ans=min(ans,cnt);
}
return ans;
}
int main()
{
int t, n, i, j, k, ans;
scanf("%d",&t);
while(t--){
scanf("%d",&n);
for(i=0;i<n;i++){
scanf("%s",mp[i]);
}
memset(a,0,sizeof(a));
for(i=0;i<n;i++){
for(j=0;j<n;j++){
for(k=0;k<4;k++){
int x=i+jx[k];
int y=j+jy[k];
if(x>=0&&x<n&&y>=0&&y<n){
a[n*i+j][n*x+y]=1;
}
}
a[n*i+j][n*i+j]=1;
if(mp[i][j]=='w'){
a[n*i+j][n*n]=1;
}
}
}
ans=gauss(n*n);
if(ans==-1){
printf("inf\n");
}
else{
printf("%d\n",ans);
}
}
return 0;
}