AcWing 95. 费解的开关 枚举 状态压缩 递推

题目描述

原题链接

分析

我们假设第$1$行的状态已经确定, 以此递推出下面$4$行的状态
由于题目需要满足第$1$行全为$1$, 所以只能通过操作第$2$行来改变第$1$行的状态, 使之全为$1$. 同时, 得到第$2$行的某种状态
同理, 由于题目需要满足第$2$行全为$1$, 所以只能通过操作第$3$行来改变第$2$行的状态, 使之全为$1$. 同时, 得到第$3$行的某种状态
同理, 由于题目需要满足第$3$行全为$1$, 所以只能通过操作第$4$行来改变第$3$行的状态, 使之全为$1$. 同时, 得到第$4$行的某种状态
同理, 由于题目需要满足第$4$行全为$1$, 所以只能通过操作第$5$行来改变第$4$行的状态, 使之全为$1$. 同时, 得到第$5$行的某种状态
同理, 由于题目需要满足第$5$行全为$1$, 所以只能通过操作第$6$行来改变第$5$行的状态, 使之全为$1$.
而由于没有第$6$行, 所以如果第5行不全为1, 则不满足要求, 需要重新确定第$1$行的状态

如何确定第$1$行的状态呢? 第$1$行每盏灯都对应按或不按两种操作, 共$2^5$种操作($0-31$)
状态压缩: 每种操作的含义可以由这$2^5$种操作对应的二进制数表示, 如00100就表示只按下第$3$栈灯
从而, 通过枚举这$2^5$种状态, 找到最优解

实现

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int n;
string line[5];
string temp[5];
int dx[] = {0,0,1,0,-1};
int dy[] = {0,1,0,-1,0};
void turn(int x, int y) // 翻转
{
    for(int i=0; i<5; i++)
    {
        int nx = x + dx[i], ny = y + dy[i];
        if(nx < 0 || nx >= 5 || ny < 0 || ny >= 5) continue; // 位置不合法
        // 将自身及上下左右共5个位置翻转(0->1, 1->0)
        temp[nx][ny] ^= 1; // 异或
    }
}
int solve()
{
    int ans = 0x3f3f3f3f;
    for(int i=0; i<32; i++) // 枚举第1行的所有操作(5个灯按或不按共2^5种), 而不是第一行灯的所有状态
    {
        int cnt = 0;
        for(int j=0; j<5; j++) temp[j] = line[j];
        // 根据对第1行的操作, 确定其状态
        for(int j=0; j<5; j++) // 状态压缩: i的二进制的第j位为1, 就表示按下该位置的灯
        {
            if((i >> j) & 1)  // 检查i的二进制的第j位是否为1
            {
                cnt++;
                turn(0,j);
            }
        }
        for(int j=0; j<4; j++) // 根据第j行的状态, 得到第j+1行的操作
        {
            for(int k=0; k<5; k++)
            {
                if(temp[j][k] == '0')
                {
                    cnt++;
                    turn(j+1,k);
                }
            }
        }
        bool flag = 1;
        // 如果第1行的操作合法, 那么递推至第5行的状态应该是全1
        for(int j=0; j<5; j++)
            if(temp[4][j] == '0')
            {
                flag = 0;
                break;
            }
        if(flag) ans = min(ans, cnt);
    }
    return ans;
}
int main()
{
    cin >> n;
    while(n--)
    {
        for(int i=0; i<5; i++) cin >> line[i];
        int ans = solve();
        if(ans > 6) ans = -1;
        cout << ans << endl;
    }
    return 0;
}