洛谷 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;
}