poj3279(二维开关问题)

/*
translation:
	给出一张二维的图表,1表示该处的电灯打开,0表示该处的电灯关闭。每次关闭一盏灯会使得其相邻的上下左右四个位置的灯状态反转。
	给出使得灯全灭的最小次数的方案以及对应的每个位置上的操作次数。
solution:
	二维开关问题。
	考虑到每次枚举左上角的灯泡,然后按照一维的开关问题求解。但是这样的话每次的状态选择就成了问题,因为上下左右四个位置都可
	使得这个位置状态发生改变。所以只需要考虑第一行的状态,每次都通过下一行来改变上一行的状态。这样每次选择的状态就成了唯一。
note:
	*:二维的开关问题解法
date:
	2016.11.7
*/
#include 
#include 
#include 

using namespace std;
const int maxn = 20;

int board[maxn][maxn];
int ans[maxn][maxn];	//保存最优解
int flip[maxn][maxn];	//记录每个位置上面的翻转次数
int n, m;	//m行n列

int dirx[] = {0, 0, 1, -1, 0};
int diry[] = {1, -1, 0, 0, 0};

bool inside(int r, int c)
{
	return r >= 0 && r < m && c >= 0 && c < n;
}

int get_color(int r, int c)
{
	int color = board[r][c];
	for(int d = 0; d < 5; d++){
		int nr = r + dirx[d];
		int nc = c + diry[d];
		if(inside(nr, nc))	color += flip[nr][nc];
	}
	return color % 2;
}

int compute()
{
	for(int i = 1; i < m; i++){
		for(int j = 0; j < n; j++){
			if(get_color(i-1, j) == 1)
				flip[i][j]++;
		}
	}

	for(int i = 0; i < n; i++)
		if(get_color(m-1, i) == 1)
			return -1;

	int res = 0;
	for(int i = 0; i < m; i++){
		for(int j = 0; j < n; j++){
			res += flip[i][j];
		}
	}
	return res;
}

int main()
{
	//freopen("in.txt", "r", stdin);
	while(~scanf("%d%d", &m, &n)){
		memset(ans, 0, sizeof(ans));
		memset(flip, 0, sizeof(flip));
		memset(board, 0, sizeof(board));

		for(int i = 0; i < m; i++){
			for(int j = 0; j < n; j++){
				scanf("%d", &board[i][j]);
			}
		}

		int cnt = -1;
		for(int i = 0; i < (1 << n); i++){
			memset(flip, 0, sizeof(flip));
			for(int j = 0; j < n; j++){
				flip[0][n-j-1] = (i >> j) & 1;
			}
			int tmp = compute();	//计算接下来的翻转次数
			if(tmp >= 0 && (cnt < 0 || tmp < cnt)){
				cnt = tmp;
				memcpy(ans, flip, sizeof(flip));
			}
		}

		if(cnt < 0){
			printf("IMPOSSIBLE\n");
			continue;
		}

		for(int i = 0; i < m; i++){
			for(int j = 0; j < n; j++)
				printf("%d%c", ans[i][j], j + 1 == n ? '\n' : ' ');
		}
	}
    return 0;
}

你可能感兴趣的:(=====杂题模型=====,开关问题)