熄灯问题学习笔记

最新推荐文章于 2024-10-21 18:03:37 发布

琉璃色的星辉

最新推荐文章于 2024-10-21 18:03:37 发布

阅读量75

点赞数

分类专栏： mooc程序设计与算法（二）学习

本文链接：https://blog.csdn.net/m0_45689014/article/details/113854543

版权

mooc程序设计与算法（二）学习专栏收录该内容

8 篇文章 0 订阅

订阅专栏

该博客讨论了一种利用二进制位运算的方法来解决开关灯问题。当第一行开关按下后，通过逐行更新灯的状态，确保上一行的灯全部熄灭，最终检查最后一行是否同样熄灭。博主提供了C++代码实现，通过枚举第一行开关的2^6种可能状态，实现了节省空间的解决方案。

摘要由CSDN通过智能技术生成

思路

就是当第一行开关按下后灯状态确定，向下逐行可以有一个确定的开关状态使得上一行的灯一定全部熄灭，直到最后一行检查是否也全部熄灭，如果是则得到结果。这里只需要枚举第一行所有开关的状态即可，更巧妙的是可以采用二进制位运算节省空间。

代码

#include <iostream>
#include <string>
#include <cstring>
using namespace std;

char oriLights[5];				//原始灯状态
char lights[5];					//改变后灯状态
char result[5];					//结果开关状态

int GetBit(char c, int i) {					//取c的第i位
	return (c >> i) & 1;				
}

void SetBit(char& c, int i, int v) {			//修改c的第i位
	if (v) {
		c |= (1 << i);
	}
	else {
		c &= ~(1 << i);
	}
}

void FlipBit(char& c, int i) {						//反转c的第i位
	c ^= (1 << i);
}

void OutputResult(int t, char result[]) {			//输出结果
	cout << "###" << t << endl;
	for (int i = 0; i < 5; ++i) {
		for (int j = 0; j < 6; ++j) {
			cout << GetBit(result[i], j);
			if (j < 5)
				cout << " ";
		}
		cout << endl;
	}
}

int main()
{
	int T;
	cin >> T;
	for (int t = 1; t <= T; ++t) {
		for (int i = 0; i < 5; ++i) 
			for (int j = 0; j < 6; ++j) {
				int s;
				cin >> s;
				SetBit(oriLights[i], j, s);
			}
		for (int n = 0; n < 64; ++n) {
			int switchs = n;									//第一行按开关策略（开始枚举，共2^6种可能）
			memcpy(lights, oriLights, sizeof(oriLights));		//初始化灯状态
			for (int i = 0; i < 5; ++i) {
				result[i] = switchs;
				for (int j = 0; j < 6; j++) {					//开关改变本行灯状态
					if (GetBit(switchs, j)) {
						if (j > 0)
							FlipBit(lights[i], j - 1);
						FlipBit(lights[i], j);
						if (j < 5)
							FlipBit(lights[i], j + 1);
					}
				}
				if (i < 4)										//第i+1行灯状态改变
					lights[i + 1] ^= switchs;
				switchs = lights[i];							//要让本行全熄灭，下一行开关状态已经确定
			}
			if (lights[4] == 0) {
				OutputResult(t, result);
				break;
			}



		}
	}

}