poj 1222 & zoj 1354 EXTENDED LIGHTS OUT (枚举)

http://poj.org/problem?id=1222

和poj 1753类似，1753是求全0或全1的步数，这题是求全0的解决方案。

当时是拿1753的代码改的，枚举步数，最多30步，这样的话状态总量就是2^30。。。

枚举第一行状态，共2^6，第一行确定了便可确定其余行，最后看末行是否全为0即可。

code:

#include<cstdio>
#include<cstring>
int map[5][6], ans[5][6] ;
int tur[5][2] = {0, 0, 0, 1, 0, -1, 1, 0, -1, 0} ;
int flag ;
void Tur(int x, int y){
for(int i=0; i<5; i++){
int tx = x + tur[i][0] ;
int ty = y + tur[i][1] ;
if(tx>=0&&ty>=0&&tx<5&&ty<6)
map[tx][ty] ^= 1 ;
}
}
void work(){ //根据第一行的状态确定其余ans
for(int i=1; i<5; i++)
for(int j=0; j<6; j++)
if(map[i-1][j]){
Tur(i, j) ;
ans[i][j] = 1 ;
}
}
void back(){
for(int i=1; i<5; i++)
for(int j=0; j<6; j++)
if(ans[i][j]){
Tur(i, j) ;
ans[i][j] = 0 ;
}
}
void test(){
flag = 1 ;
for(int i=0; i<6; i++)
if(map[4][i]==1) flag = 0 ;
}
void print(){
int i, j ;
for(i=0; i<5; i++){
for(j=0; j<5; j++)
printf("%d ", ans[i][j]) ;
printf("%d\n", ans[i][j]) ;
}
}
void dfs(int y){ //枚举第一行64种状态
if(flag) return ;
if(y==6){
work() ;
test() ;
if(flag) print() ;
else back() ;
}
else{
Tur(0, y) ;
ans[0][y] = 1 ;
dfs(y+1) ;
Tur(0, y) ;
ans[0][y] = 0 ;
dfs(y+1) ;
}
}
int main(){
int t=0, i, j, n ;
scanf("%d", &n) ;
while(n--){
memset(ans, 0, sizeof(ans)) ;
flag = 0 ;
for(i=0; i<5; i++)
for(j=0; j<6; j++)
scanf("%d", &map[i][j]) ;
printf("PUZZLE #%d\n", ++t) ;
dfs(0) ;
}
return 0 ;}