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

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

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

具体如下：

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

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

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

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

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

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

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

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

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

开始时所有灯都是熄灭的，开关是关闭着的。要求按下若干开关后，使得只有4盏灯亮着。
输入
无
输出
输出所有可能的方案，每行一个方案，每一行有9个字符，从左往右第i个字符表示第i个开关的状态(" 0" 表示关闭，" 1" 表示打开)，按字典序输出。下面的样例输出只是部分方案。
样例输入
无
样例输出
000001011
000001110
000001111
思路：数据量不大，直接暴力搜索，然后判断是否符合题意。注意字典序是从小到大。
代码如下：

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const int maxx=15;
int vis[maxx];
int a[maxx];

inline int check()
{
	memset(a,0,sizeof(a));
	if(vis[1]) a[2]++,a[4]++;
	if(vis[2]) a[1]++,a[3]++,a[5]++;
	if(vis[3]) a[2]++,a[6]++;
	if(vis[4]) a[1]++,a[5]++,a[7]++;
	if(vis[5]) a[2]++,a[4]++,a[6]++,a[8]++;
	if(vis[6]) a[3]++,a[5]++,a[9]++;
	if(vis[7]) a[4]++,a[8]++;
	if(vis[8]) a[5]++,a[7]++,a[9]++;
	if(vis[9]) a[6]++,a[8]++;
	int num=0;
	for(int i=1;i<=9;i++) if(a[i]&1) num++;
	return num==4;
}
inline void dfs(int x)
{
	if(x==10)
	{
		if(check())
		{
			for(int i=1;i<=9;i++) cout<<vis[i];
			cout<<endl;
		}
		return ;
	}
	dfs(x+1);//先考虑不开的情况，这样字典序就是最小的了。
	if(vis[x]==0) 
	{
		vis[x]=1;
		dfs(x+1);
		vis[x]=0;
	}
}
int main()
{
	memset(vis,0,sizeof(vis));
	dfs(1);
	return 0;
}

努力加油a啊，(^o)/~