费解的开关(BFS+状态压缩)

PS:代码超时的,可惜。正解好像是DP,很好奇为什么DFS能卡过而BFS不行

题目链接费解的开关

题目描述

你玩过“拉灯”游戏吗?25盏灯排成一个5x5的方形。每一个灯都有一个开关,游戏者可以改变它的状态。每一步,游戏者可以改变某一个灯的状态。游戏者改变一个灯的状态会产生连锁反应:和这个灯上下左右相邻的灯也要相应地改变其状态。

我们用数字“1”表示一盏开着的灯,用数字“0”表示关着的灯。下面这种状态

10111
01101
10111
10000
11011

在改变了最左上角的灯的状态后将变成:

01111
11101
10111
10000
11011

再改变它正中间的灯后状态将变成:

01111
11001
11001
10100
11011

给定一些游戏的初始状态,编写程序判断游戏者是否可能在6步以内使所有的灯都变亮。

输入格式

第一行输入正整数n,代表数据***有n个待解决的游戏初始状态。

以下若干行数据分为n组,每组数据有5行,每行5个字符。每组数据描述了一个游戏的初始状态。各组数据间用一个空行分隔。

输出格式

一共输出n行数据,每行有一个小于等于6的整数,它表示对于输入数据中对应的游戏状态最少需要几步才能使所有灯变亮。

对于某一个游戏初始状态,若6步以内无法使所有灯变亮,则输出“-1”。

数据范围

0<n≤500

输入样例:
3
00111
01011
10001
11010
11100

11101
11101
11110
11111
11111

01111
11111
11111
11111
11111
输出样例:
3
2
-1

思路

反向bfs,因为我们要搜的是搜索树中的其中一个边界,而我们得到的只是状态搜索树的其中一个节点,bfs的话可能会把双边的情况都考虑到,这样就可能爆内存,而从结果反向搜起点就没有这种多余的枚举。用map把重复的状态删除,这个图的状态可以压缩再一个25位的整数中。只可惜TLE了

C++ 代码

#include <bits/stdc++.h>

using namespace std;
int dir[4] = {1,-1,5,-5};
int num;
map<int,int> mp;
struct Node {
	int step;
	int now;
	
	Node(){}
	
	Node (int step, int now) {
		this->step = step;
		this->now = now;
	}
	
	Node operator = (const Node& A) {
		step = A.step;
		now = A.now;
		return *(this);
	}
	
	void change (int idx) { //灯状态变换
		this->now ^= (1 << (25 - idx));
		for (int i = 0; i < 4; i++) {
			int op = idx + dir[i];
			if (op < 1 || op > 25 || abs(idx-op) == 1 && abs(idx+op) % 10 == 1) {
				continue;
			}
			this->now ^= (1 << (25 - op));
		}
		
		this->step++;
	}
	
	bool isok () {
		if (now == num) {
			return true;
		}
		return false;
	}
	
};

queue<Node> Q;
int BFS (int s) {
	Node ths(0, s),nxt;
	while (!Q.empty()) {
		Q.pop();
	}
	mp.clear();
	mp.insert(make_pair(s,1));
	Q.push(ths);
	while (!Q.empty()) {
		ths = Q.front();
		Q.pop();
		for (int i = 1; i <= 25; i++) {
			nxt = ths;
			nxt.change(i);
			if (nxt.isok()) {
				return nxt.step;
			} else if (nxt.step < 6 && mp.find(nxt.now) == mp.end()){
				Q.push(nxt);
				mp.insert(make_pair(nxt.now, 1));
			}
		}
	}
	return -1;
}

int main() {
	int T;
	scanf("%d", &T);
	while (T--) {
		num = 0;
		for (int i = 0; i < 5; i++) {
			getchar();
			for (int j = 0; j < 5; j++) {
				char ch = getchar();
				num <<= 1;
				num |= (ch - '0');
			}
		}
		getchar();
		printf("%d\n",BFS((1 << 25)-1));
	}
	return 0;
}
全部评论

相关推荐

上周组里招人,我面了六个候选人,回来跟同事吃饭的时候聊起一个让我挺感慨的现象。前三个候选人,算法题写得都不错。第一道二分查找,五分钟之内给出解法,边界条件也处理得干净。第二道动态规划,状态转移方程写对了,空间复杂度也优化了一版。我翻他们的简历,力扣刷题量都在300以上。后三个呢,就有点参差不齐了。有的边界条件没处理好,有的直接说这道题没刷过能不能换个思路讲讲。其中有一个女生,我印象特别深——她拿到题之后没有马上写,而是先问我:“面试官,我能先跟你确认一下我对题目的理解吗?”然后她把自己的思路讲了一遍,虽然最后代码写得不是最优解,但整个沟通过程非常顺畅。这个女生的代码不是最优的,但当我问她“如果这里是线上环境,你会怎么设计’的时候,她给我讲了一套完整的方案——异常怎么处理、日志怎么打、怎么平滑发布。她对这是之前在实习的时候踩过的坑。”我在想LeetCode到底在筛选什么?我自己的经历可能有点代表性。我当年校招的时候,也是刷了三百多道题才敢去面试。那时候大家都刷,你不刷就过不了笔试关。后来工作了,前三年基本没再打开过力扣。真正干活的时候,没人让你写反转链表,也没人让你手撕红黑树。更多的是:这个接口为什么慢了、那个服务为什么OOM了、线上数据对不上了得排查一下。所以后来我当面试官,慢慢调整了自己的评判标准。算法题我还会出,但目的变了。我出算法题,不是想看你能不能背出最优解。而是想看你拿到一个陌生问题的时候,是怎么思考的。你会先理清题意吗?你会主动问边界条件吗?你想不出来的时候会怎么办?你写出来的代码,变量命名乱不乱、结构清不清楚?这些才是工作中真正用得到的能力。LeetCode是一个工具,不是目的。它帮你熟悉数据结构和常见算法思路,这没问题。但如果你刷了三百道题,却说不清楚自己的项目解决了什么问题、遇到了什么困难、你是怎么解决的,那这三百道题可能真的白刷了。所以还要不要刷LeetCode?要刷,但别只刷题。刷题的时候,多问自己几个为什么:为什么用这个数据结构?为什么这个解法比那个好?如果换个条件,解法还成立吗?把刷题当成锻炼思维的方式,而不是背答案的任务。毕竟面试官想看到的,从来不是一台背题机器,而是一个能解决问题的人。
牛客51274894...:意思是光刷力扣还不够卷
AI时代还有必要刷lee...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务