【POJ 1222】EXTENDED LIGHTS OUT | 状压 | 枚举 | N

                     【POJ 1222】EXTENDED LIGHTS OUT

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 13670		Accepted: 8563

URL

【POJ 1222】EXTENDED LIGHTS OUT

Description

给定一个 5 行 6 列矩阵，0 代表熄灭的灯，1代表打开的灯。

对于每盏灯可以进行按下操作，按下后这盏灯自己以及它上下左右的灯（如果存在的话）的开关状态会反转。

现在需要给出一个操作集，使得按下某些灯后，所有的灯都熄灭。

下图是按下三个标记位置的灯的情况。

下图也是按下三个标记位置的灯的情况，注意这三次操作后第一行的灯都关掉了。

题目给的一些提示：

1. 灯关闭的顺序不重要
2. 灯连续按两次的话相当于没按，所以对于每盏灯，要么按一次，要么不按。也就是说输出的操作集也是一个 01 矩阵
3. 编程的思路提示：参见上面的图，第二行这样操作后第一行的灯就都关掉了，那么在第三行再进行某些操作就可以把第二行都关掉，依次类推直到最后一行。当然也可以按列来进行，先对第二列进行某些操作关掉第一列，再对第三列进行某些操作关掉第二列，再....直到最后一列。

Input

第一行，一个整数 n，代表数据组数。

对于每组数据，输入一个 6 行 5 列的 01 矩阵代表灯的初始状态。

Output

输出对应的操作集（也是一个 6 行 5 列的 01 矩阵），0代表不按对应位置的灯，1代表按。

输出每组数据前输出一个PUZZLE #i（i从1到n）

Sample Input

2
0 1 1 0 1 0
1 0 0 1 1 1
0 0 1 0 0 1
1 0 0 1 0 1
0 1 1 1 0 0
0 0 1 0 1 0
1 0 1 0 1 1
0 0 1 0 1 1
1 0 1 1 0 0
0 1 0 1 0 0

Sample Output

PUZZLE #1
1 0 1 0 0 1
1 1 0 1 0 1
0 0 1 0 1 1
1 0 0 1 0 0
0 1 0 0 0 0
PUZZLE #2
1 0 0 1 1 1
1 1 0 0 0 0
0 0 0 1 0 0
1 1 0 1 0 1
1 0 1 1 0 1

Analysis

这道题可以就按照题目给的提示来做，也就是逐行分析。

可以发现，当第一行的操作情况确定后，其他行的操作情况也就唯一确定了。下面进行证明：

 

比如第二行的操作就必须把第一行还没有关闭的灯都关掉。也就是说第一行的哪些列的灯没关，第二行就必须按下这些列的开关把他们关掉，并且不能再按其他的灯，否则第一行本来关的灯又打开了。

为什么第二行一定要这样做才能保证最终第一行的灯都关掉呢？因为：

• 第一行已经结束操作，不能再改变自己的状态。
• 第三行及后面的行无法通过操作来改变第一行的灯的状态（只能改变相邻行的状态）

所以只能通过第二行的操作来关掉第一行还没有关闭的灯。同理，第二行操作结束后，也只能通过第三行的操作关掉第二行还没有关掉的灯...一直到最后一行。也就是说每一行的操作都直接由它上一行操作后的状态决定。

【也就是说第 2~5 行的操作都直接由第一行操作后的状态决定！】

这样就好办了。直接枚举第一行的所有操作情况（也就 2^6 == 64 种），然后逐行进行操作，全部完成之后检查最后一行有没有全关掉，如果最后一行全关掉了，那么所有的灯就全关掉了（因为之前的操作确保了上一行的灯会全部关掉），就找到了解。

 

当然，这种反转状态的操作以及01 状态很容易让人想到状压。此题用状压也是很方便的，速度也会得到提升，因此我们可以把每一行都压缩成一个 unsigned char 来做。

 

AC code

代码如下，关键的地方有相应的注释：

#include <cstdio>
#include <cstring>

const int R = 5;
const int C = 6;

typedef unsigned char Line;

Line state0[R], line[R], operation[R];

inline bool getBit(Line c, unsigned k)
{
	return (c >> k) & 1;
}

inline void setBit(Line &c, unsigned k, bool v)
{
	if (v)
		c |= (1 << k);
	else
		c &= ~(1 << k);
}

inline void flipBit(Line &c, unsigned k)
{
	c ^= (1 << k);
}

void println(Line op)
{
	for (int i=0; i<C; ++i, op>>=1)
		if (op & 1)
			putchar('1');
		else
			putchar('0');
		putchar(' ');
	putchar('\n');
}

int main()
{
	int T, value;
	scanf("%d", &T);
	for (int t=1; t<=T; ++t)
	{
		printf("PUZZLE #%d\n", t);

		for (int r=0; r<R; ++r)
		{
			for (int c=0; c<C; ++c)
			{
				scanf("%d", &value);
				setBit(state0[r], c, value);
			}
		}

		// 枚举第一行的操作
		for (operation[0]=0; operation[0]<64; ++operation[0])
		{
			line[0] = state0[0];
			
			// 进行每行的操作 
			for (int r=0; r<R; ++r)
			{
				// 先更新当前行的灯的状态 
				if (getBit(operation[r], 0))	// 是否按下当前行最左边的灯
				{
					flipBit(line[r], 0);
					flipBit(line[r], 1);
				}

				for (int c=1; c<C-1; ++c)	// 是否按下当前行中间区域的灯
				{
					if (getBit(operation[r], c))
					{
						flipBit(line[r], c-1);
						flipBit(line[r], c);
						flipBit(line[r], c+1);
					}
				}

				if (getBit(operation[r], C-1))	// 是否按下当前行最右边的灯
				{
					flipBit(line[r], C-1);
					flipBit(line[r], C-2);
				}

				// 再更新当前行操作对下一行的影响（不必再更新对上一行的影响，因为经过这一行操作后上一行一定是全灭
				// 再根据当前行操作后的状态，确定下一行的操作
				if (r+1 < R)
				{
					operation[r+1] = line[r];
					line[r+1] = state0[r+1] ^ operation[r];
				}
			}

			// 如果最后一行也全灭了，说明每一行都灭了，找到了解。 
			if (!line[R-1])
			{
				for (int r=0; r<R; ++r)
					println(operation[r]);
				break;
			}
		}
	}

	return 0;
}