UVA 10318 - Security Panel dfs 剪枝

UVA 10318 - Security Panel dfs 剪枝

ACM

题目地址：UVA 10318 - Security Panel

题意： 
这题跟点灯的题目很像，点灯游戏选择一盏灯时会让它以及四周的灯改变状态。 
但是我们有特殊的开开关技巧，它给出了改变状态的位置，而不是四周都改变。 
问你从全部关着变成全部开着的最小开关步骤。

分析： 
很明显，在一个位置上点两次或更多次是没有必要的，所以一个位置只有选择与不选择，用dfs即可，但如果暴力所有可能，复杂度是2^25，会超时，所以要剪枝。 
由于影响范围是3*3的，所以开关是无法影响两行前的状态的，只要在要走向下一行前判断下就行了。 
由于数据量很小，这个剪枝足够了。

代码：

/*
*  Author:      illuz <iilluzen[at]gmail.com>
*  File:        10318.cpp
*  Create Date: 2014-06-29 19:51:39
*  Descripton:   
*/

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int dx[] = {-1,-1,-1, 0, 0, 0, 1, 1, 1};
const int dy[] = {-1, 0, 1,-1, 0, 1,-1, 0, 1};

const int N = 6;

int r, c, cas;
int change[3*3], lit[N][N], prs[N*N], ans[N*N], cnt;
char str[4];

bool check(int l1) {
	if (l1 >= 0) {
		for (int i = 0; i < c; i++) {
			if (!lit[l1][i]) {
				return false;
			}
		}
	}
	return true;
}

void pressfunc(int x, int y) {
	for (int i = 0; i < 9; i++) {
		if (change[i]) {
			int tx = x + dx[i], ty = y + dy[i];
			if (tx >= 0 && tx < r && ty >= 0 && ty < c) {
				lit[tx][ty] = !lit[tx][ty];
			}
		}
	}
}

void dfs(int x, int y, int press) {
	if (y == c) {
		if (x == r - 1) {
			// check if all light
			for (int i = 0; i < r; i++) {
				if (!check(i)) {
					return;
				}
			}

			// if all light, update the ans
			if (press < cnt) {
				cnt = press;
				for (int i = 0; i < press; i++) {
					ans[i] = prs[i];
				}
			}
			return;
		} else {
			if (!check(x - 2)) {	// if not all light in line x - 2
				return;
			}
			x++;
			y = 0;
		}
	}

	// not press
	dfs(x, y + 1, press);

	// press
	pressfunc(x, y);
	prs[press] = c * x + y + 1;
	dfs(x, y + 1, press + 1);
	pressfunc(x, y);	// roll back
}

int main() {
	while (~scanf("%d%d", &r, &c) && (r || c)) {
		// input
		for (int i = 0; i < 3; i++) {
			scanf("%s", str);
			for (int j = 0; j < 3; j++) {
				change[i*3+j] = str[j] == '*' ? 1 : 0;
			}
		}

		// init
		memset(lit, 0, sizeof(lit));
		cnt = r * c + 1;

		// solve
		printf("Case #%d\n", ++cas);
		dfs(0, 0, 0);
		if (cnt != r * c + 1) {
			for (int i = 0; i < cnt - 1; i++) {
				printf("%d ", ans[i]);
			}
			printf("%d\n", ans[cnt - 1]);
		} else {
			puts("Impossible.");
		}
	}
	return 0;
}