洛谷 P2622 关灯问题II

题目来源：https://www.luogu.org/problem/show?pid=2622

状态压缩bfs。注意判重。

代码：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <sstream>
#include <cstdio>
#include <string>
#include <vector>
#include <cmath>
#include <ctime>
#include <stack>
#include <queue>
#include <set>
#include <map>
using namespace std;
int a[101][101]={0};
int main(){
ios::sync_with_stdio(false);
int n,m;
cin>>n>>m;
for(int i=1;i<=m;i++)for(int j=1;j<=n;j++)cin>>a[i][j];
set<int> s;
queue<int> q,p;
int t=0;
for(int i=0;i<n;i++)t=((1<<i)|t);
s.insert(t);
p.push(t);
int tot=0;
while(!p.empty())
{
int l=p.front();p.pop();
if(l==0){cout<<tot;return 0;}
for(int i=1;i<=m;i++)
{
int w=l;
for(int j=1;j<=n;j++)
{
if(a[i][j]==1)
{
if(((1<<(j-1))&l)!=0)w-=pow(2,j-1);
}
if(a[i][j]==-1)
{
int u=l+pow(2,j-1);
if(((1<<(j-1))&u)!=0)w+=pow(2,j-1);
}
}
if(!s.count(w))
{
s.insert(w);q.push(w);
}
}
if(p.empty())
{
while(!q.empty())
{
p.push(q.front());q.pop();
}
tot++;
}
}
cout<<"-1";
return 0;
}