洛谷 P2622 关灯问题II (状压bfs)

P2622 关灯问题II

luogu链接

题目描述

现有n盏灯，以及m个按钮。每个按钮可以同时控制这n盏灯——按下了第i个按钮，对于所有的灯都有一个效果。按下i按钮对于第j盏灯，是下面3中效果之一：如果a[i][j]为1，那么当这盏灯开了的时候，把它关上，否则不管；如果为-1的话，如果这盏灯是关的，那么把它打开，否则也不管；如果是0，无论这灯是否开，都不管。

现在这些灯都是开的，给出所有开关对所有灯的控制效果，求问最少要按几下按钮才能全部关掉。

输入格式

前两行两个数，n m

接下来m行，每行n个数,a[i][j]表示第i个开关对第j个灯的效果。

输出格式
一个整数，表示最少按按钮次数。如果没有任何办法使其全部关闭，输出-1

输入输出样例
输入 #1
3
2
1 0 1
-1 1 0
输出 #1
2
说明/提示
对于20%数据，输出无解可以得分。

对于20%数据，n<=5

对于20%数据，m<=20

上面的数据点可能会重叠。

对于100%数据 n<=10,m<=100

Solution

经典状态压缩
隐式图转化为显式图通过BFS最短路解决
起点为全开 终点为全关
状态最多 1 << 10 - 1

代码

#include<bits/stdc++.h>
using namespace std;
const int SZ = 100 + 5;
int a[SZ][12],n,m,ans;
struct zt
{
	int sta;
	int step;
};
int vis[2049];
inline void bfs(int t)
{
	ans = -1; 
	queue<zt> q;
	q.push((zt){t,0});
	vis[t] = 1;
	while(!q.empty())
	{
		zt now = q.front();
		q.pop(); 
		if(now.sta == 0) //到达终点 
		{
			ans = now.step;
			break;
		} 
		for(int i = 1;i <= m;i ++ ) //枚举每个开关对所有灯的影响 
		{
			int skr = now.sta;
			for(int j = 1;j <= n;j ++ ) //一个开关要把所有灯的状态都变化 
			{
				if(a[i][j] == 1 && ((1 << (n - j)) & skr))
					skr = skr ^ (1 << (n - j));
				 if(a[i][j] == -1)
					skr = skr |  (1 << (n - j));
			}
			if(vis[skr] == 0) //防止重复入队 
			{
				q.push((zt){skr,now.step + 1});
				vis[skr] = 1; 
			}	
		} 
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i = 1;i <= m;i ++ ) // n和m搞反 wa了好几发 
		for(int j = 1;j <= n;j ++ )
			scanf("%d",&a[i][j]);
	int N = (1 << n) - 1;
	bfs(N);
	if(ans == -1) printf("-1\n");
	else printf("%d\n",ans);
	return 0;
}

2020.4.8