[蓝桥杯][算法提高VIP]开灯游戏

题目描述

有9盏灯与9个开关，编号都是1~9。 

每个开关能控制若干盏灯，按下一次会改变其控制的灯的状态(亮的变成不亮，不亮变成亮的)。 

具体如下： 

第一个开关控制第二，第四盏灯； 

第二个开关控制第一，第三，第五盏灯； 

第三个开关控制第二，第六盏灯； 

第四个开关控制第一，第五，第七盏灯； 

第五个开关控制第二，第四，第六，第八盏灯； 

第六个开关控制第三，第五，第九盏灯； 

第七个开关控制第四，第八盏灯； 

第八个开关控制第五，第七，第九盏灯； 

第九个开关控制第六，第八盏灯。 

开始时所有灯都是熄灭的，开关是关闭着的。要求按下若干开关后，使得只有4盏灯亮着。 

输入

无

输出

输出所有可能的方案，每行一个方案，每一行有9个字符，从左往右第i个字符表示第i个开关的状态(" 0" 表示关闭，" 1" 表示打开)，按字典序输出。下面的样例输出只是部分方案。 

样例输入

无

样例输出

000001011 
000001110 
000001111 

#include<iostream>
#include<set>
#include<vector>
using namespace std;
vector<int> op[10];
set<vector<int> > ans;
vector<int> show(10,0),state(10,0);
bool judge(vector<int> & show){
	int cnt = 0;
	for(int i = 1;i <= 9;i ++){
		if(show[i] == 1) cnt ++; 
	}
	return cnt == 4;
}

void dfs(int idx,bool press){
	if(idx == 10){
		if(judge(show)){
			ans.insert(state);
		}
		return;
	}
	if(press){
		state[idx] = 1;
		for(int i = 0;i < op[idx].size();i ++){
			show[op[idx][i]] ^= 1;
		}
		dfs(idx+1,1);
		dfs(idx+1,0);
		state[idx] = 0;
		for(int i = 0;i < op[idx].size();i ++){
			show[op[idx][i]] ^= 1;
		}
	}
	else{
		dfs(idx+1,1);
		dfs(idx+1,0);
	}
}
int main()
{
	op[1].push_back(2);
	op[1].push_back(4);
	
	op[2].push_back(1);
	op[2].push_back(3);
	op[2].push_back(5);
	
	op[3].push_back(2);
	op[3].push_back(6);
	
	op[4].push_back(1);
	op[4].push_back(5);
	op[4].push_back(7);
	
	op[5].push_back(2);
	op[5].push_back(4);
	op[5].push_back(6);
	op[5].push_back(8);
	
	op[6].push_back(3);
	op[6].push_back(5);
	op[6].push_back(9);
	
	op[7].push_back(4);
	op[7].push_back(8);
		
	op[8].push_back(5);
	op[8].push_back(7);
	op[8].push_back(9);
	
	op[9].push_back(6);
	op[9].push_back(8);
	
	dfs(0,0);
	for(set<vector<int> >::iterator it = ans.begin();it != ans.end();it ++){
		vector<int> tmp = *it;
		for(int i = 1;i <= 9;i ++){
			cout<<tmp[i];
		}
		cout<<endl;
	}
	return 0;
}