POJ 1222 : EXTENDED LIGHTS OUT

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

点灯问题，用遍历搜索空间的方法解。正如题目中所提到的几点，按下的顺序无关，每个灯最多只需按下一次。假设最终的按下位置为press矩阵，那么如果press矩阵是寻找的答案，那么按照press的第一行矩阵对灯进行操作之后，此时在矩阵的第一行中：

如果位置(1,j)上的灯是亮的，则要按下位置(2,j)，即press[2][j] 一定取1；

如果位置(1,j)上的灯是熄灭的，则不能按位置(2,j)上的按钮，press[2][j] 一定取0。

这样一句位置press的第一，二行操作矩阵中的按钮，才能保证第一行的灯全部熄灭。而对矩阵中的第三四五行无论怎么操作都不影响第一行的灯。以此类推，可以确定第三四五行中的press值。当第一行的值确定时，其他行的值也必定确定。而第一行的取值有2^6种情况，这样解空间就容易比较小。（以上参考《程序设计导引及在线实践》）程序如下：

/*
ID: csuchenan
PROG:POJ 1222
LANG: C++
Algorithm: complete search
*/

#include<iostream>
#include<stdio.h>
#include<string.h>

using namespace std ;

int puzzle[6][8] ;
int press[6][8] ;

bool judge()  ;
void comsea() ;

int main()
{
	int tests ;
	int cases ;
	
	scanf("%d" , &tests) ;
	
	int i ;
	int j ;
	cases = 1 ;
	while(cases <= tests )
	{
		memset(press , 0 , sizeof(press)) ;
		memset(puzzle , 0 , sizeof(puzzle)) ;
		
		for(i = 1 ; i < 6 ; i ++)
			for(j = 1 ; j < 7 ; j ++)
				scanf("%d" , &puzzle[i][j]) ;
		
		comsea() ;
		
		printf("PUZZLE #%d\n" , cases ++) ;
		
		for(i = 1 ; i < 6 ; i ++)
		{
			for(j = 1 ; j < 7 ; j ++)
			{	
				if(j != 6)
					printf("%d " , press[i][j]) ;
				else 
					printf("%d" , press[i][j]) ;
			}
			printf("\n") ;	
		}
	}
}

void comsea()
{
	int i ;
	
	while(judge()== false)
	{
		press[1][1] ++ ;
		
		i = 1 ;
		
		while(press[1][i] > 1 && i < 7)
		{
			press[1][i] = 0 ;
			i ++ ;
			press[1][i] ++ ;
		}
	}
}

bool judge()
{
	int i ;
	int j ;
	
	for(i = 1 ; i < 5 ; i ++)
	{
		for(j = 1 ; j < 7 ; j ++)
		{
			press[i+1][j] = (puzzle[i][j] + press[i][j] + press[i][j-1] + press[i][j+1] + press[i-1][j]) % 2 ;
		}
	}
	
	for(i = 1 ; i < 7 ; i ++)
	{
		if((press[5][i-1] + press[5][i] + press[5][i+1] + press[4][i])%2 != puzzle[5][i]) 
				return false ;
	}
	
	return true ;
}