题解AcWing95费解的开关

Posted 2021-03-09 xsl19

题面

一道非常好的递推练习题。

我们考虑每次枚举第一行的操作，由上一行的状态递推出下一行的状态，最后判断最后一行是否全部为 (1) 即可。

实现代码时要注意一些细节问题。

#include <bits/stdc++.h>
#define DEBUG fprintf(stderr, "Passing [%s] line %d
", __FUNCTION__, __LINE__)
#define itn int
#define gI gi

using namespace std;

typedef long long LL;
typedef pair <int, int> PII;
typedef pair <int, PII> PIII;

inline int gi()
{
    int f = 1, x = 0; char c = getchar();
    while (c < '0' || c > '9') {if (c == '-') f = -1; c = getchar();}
    while (c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return f * x;
}

inline LL gl()
{
    LL f = 1, x = 0; char c = getchar();
    while (c < '0' || c > '9') {if (c == '-') f = -1; c = getchar();}
    while (c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return f * x;
}

const int dx[] = {0, -1, 0, 1, 0}, dy[] = {0, 0, 1, 0, -1}; 
//定义常量数组表示当前要翻转的灯及它上下左右的位置

int n, m;
char g[7][7];

inline void get(int x, int y) //摁下第 x 行 y 列的灯
{
    for (int i = 0; i < 5; i+=1) //将它本身及四周的灯翻转状态
    {
        int xx = x + dx[i], yy = y + dy[i];
        if (xx >= 0 && xx <= 4 && yy >= 0 && yy <= 4) //在界内
        {
            //进行翻转
            if (g[xx][yy] == '1') g[xx][yy] = '0';
            else g[xx][yy] = '1';
        }
    }
}

inline int getans() //求答案
{
    int ans = 66666666;
    for (int k = 0; k < (1 << 5); k+=1) //枚举每一种第一行的操作状态
    {
        int sum = 0; //操作的总数
        char bf[7][7];
        memcpy(bf, g, sizeof g); //先将初始状态备份
        for (int j = 0; j < 5; j+=1)
        {
            if (k >> j & 1) //如果第 j 号灯需要翻转
            {
                ++sum; //操作一次
                get(0, j); //将第 j 号灯翻转
            }
        }
        for (int i = 0; i < 4; i+=1) //递推出下一行的状态
        {
            for (int j = 0; j < 5; j+=1) //枚举第 i 行的每一盏灯
            {
                if (g[i][j] == '0') //如果这个灯需要被翻转
                {
                    ++sum; 
                    get(i + 1, j); //将它下一排的灯翻转
                }
            }
        }
        bool ok = true;
        for (int j = 0; j < 5; j+=1)
        {
            if (g[4][j] == '0') {ok = false; break;} //没有达到目标状态
        }
        if (ok) ans = min(ans, sum); //记录最少步数
        memcpy(g, bf, sizeof g); //还原备份
    }
    if (ans > 6) return -1; //不能在 6 步以内达到目标状态
    return ans; //返回答案
}

int main()
{
    //freopen(".in", "r", stdin);
    //freopen(".out", "w", stdout);
    int t = gi(); //多组数据输入数据总数
    while (t--)
    {
        for (int i = 0; i < 5; i+=1) scanf("%s", g[i]); //输入每一行的状态
        printf("%d
", getans()); //输出答案
    }
    return 0;
}