POJ 1222 EXTENDED LIGHTS OUT 高斯消元？暴力枚举！

题目链接：http://poj.org/problem?id=1222

题目大意：给一个6*5的矩阵，表示30个灯泡的开关情况，其中0代表关，1代表开。可以认为每个灯上有一个开关，按下开关之后，那个灯以及上下左右的4个灯的开关状态会发生改变。求让所有的灯熄灭的解决方法，最后输出5*6的矩阵，0表示不按那个按钮，1表示按。

 

一开始听说高斯消元什么的……后来发现没有必要，枚举第一行灯泡的开关情况，然后第2、3、4、5行为了让上一行保持熄灭，都是固定的，最后判断一下最后一行是不是也熄灭就行了。说起来就是模拟+DFS，最多搜2^6次，很容易就过了。

 

AC代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
using namespace std;

int map[6][5];
int ans[6][5];
int buffer[6][5];

bool DFS(int x,int pos)
{
	int i,j;
	
	if(pos==4)
	{
		for(i=0;i<6;i++)
			ans[i][pos]=buffer[i][pos-1];
		for(i=0;i<6;i++)
		if(ans[i][pos])
		{
			if(i>=1)
			buffer[i-1][pos]=buffer[i-1][pos]^1;//改变
			buffer[i][pos]= buffer[i][pos]^1;
			if(i<5)
			buffer[i+1][pos]= buffer[i+1][pos]^1;
		}
		for(j=0;j<6;j++)
		if(buffer[j][pos])
		{
			for(i=0;i<6;i++)
			if(ans[i][pos])
			{
				if(i>=1)
				buffer[i-1][pos]=buffer[i-1][pos]^1;//改变
				buffer[i][pos]= buffer[i][pos]^1;
				if(i<5)
				buffer[i+1][pos]= buffer[i+1][pos]^1;
			}
			return false;	
		}
		return true;
	}
	
	if(x==6)
		return DFS(0,pos+1);
	 
	if(pos==0 || buffer[x][pos-1]==1)
	{
		ans[x][pos]=1;
		if(x>=1)
		buffer[x-1][pos]=buffer[x-1][pos]^1;//改变
		buffer[x][pos]= buffer[x][pos]^1;
		if(x<5)
		buffer[x+1][pos]= buffer[x+1][pos]^1;
		buffer[x][pos+1]=buffer[x][pos+1]^1;
		
		if(DFS(x+1,pos)) return true;
		
		if(x>=1)
		buffer[x-1][pos]=buffer[x-1][pos]^1;//变回 
		buffer[x][pos]= buffer[x][pos]^1;
		if(x<5)
		buffer[x+1][pos]= buffer[x+1][pos]^1;
		buffer[x][pos+1]=buffer[x][pos+1]^1;
	}
	if(pos==0 || buffer[x][pos-1]==0) 
	{
		ans[x][pos]=0;
		if(DFS(x+1,pos)) return true;
	}
	
	return false;
}

int main()
{
//	freopen("out.txt","w",stdout);
	int i,j,a,T,tot=0;
	cin>>T;
	while(T--)
	{
		for(j=0;j<5;j++)
		for(i=0;i<6;i++)
		{
			scanf("%d",&map[i][j]);
			buffer[i][j]=map[i][j];
		}
		if(!DFS(0,0)) cout<<"woc?"<<endl; 
		cout<<"PUZZLE #"<<++tot<<endl;
		for(j=0;j<5;j++)
		{
			for(i=0;i<6;i++)
			printf("%d ",ans[i][j]);
			cout<<endl;
		}
		
	}
}