题目链接: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;
}
}
}