洛谷-P2262-关灯问题II（bfs + 状态压缩初探）

题目描述

现有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

解题思路：

这题首先可以想到是搜索，bfs，枚举n个开关对每个状态的影响。
那么对于本题中的状态，可以知道，每个点只有两个状态，即开或者关，那么我们可以用二进制中的数位来表示一个灯的开关，也就是说对于第i个灯，如果第i位上的数字是i那么就说这个灯是开着的，否则就是关着的。
那么修改一个状态的话也可以利用位运算的性质，对一个数的第i位进行修改。
于是代码就很好写了，用二进制表示状态然后dfs搜索，搜索到状态值为0时就说明找到了解决方案。
其实说是状态压缩，其实也没有压缩啊，无非就是用二进制数表示一个状态值。说白了就是简单的暴力。

AC代码：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e6 +10;
struct node {
    int sta,step;
    node(int a,int b)
    {
        sta = a;
        step = b;
    }
};

int a[105][105];
int n,m;
bool mark[1<<11];

int bfs()
{
    int sta = 1;
    for(int i = 1 ; i < m ; i ++)
    {
        sta |= 1<<i;
    }
    memset(mark,0,sizeof(mark));
    queue<node> que;
    que.push(node(sta,0));
    while(!que.empty())
    {
        node tmp = que.front();
        que.pop();
        if(tmp.sta == 0)
            return tmp.step;
        for(int i = 0 ; i < n ; i ++)
        {
            int tmp1 = tmp.sta;
            for(int j = 0 ; j < m ; j ++)
            {
                if(a[i][j] == 1 && (tmp.sta&(1<<j)))
                    tmp1 = tmp1^(1<<j);
                if(a[i][j] == -1)
                    tmp1 = tmp1|(1<<j);
            }
            if(!mark[tmp1])
                que.push(node(tmp1,tmp.step+1));
            mark[tmp1] = true;
        }
    }
    return -1;
}

signed main()
{
    cin>>m>>n;
    for(int i = 0 ; i < n ; i ++)
        for(int j = 0 ; j < m ; j ++)
            cin>>a[i][j];
    cout<<bfs()<<endl;
}