开关灯泡问题

目录

题目描述

解法

---

链接：95. 费解的开关 - AcWing题库

题目描述

25盏灯排成一个 5×5 的方形。

每一个灯都有一个开关，游戏者可以改变它的状态。

每一步，游戏者可以改变某一个灯的状态。

游戏者改变一个灯的状态会产生连锁反应：和这个灯上下左右相邻的灯也要相应地改变其状态。

我们用数字 1 表示一盏开着的灯，用数字 0 表示关着的灯。

给定一些游戏的初始状态，编写程序判断游戏者是否可能在 66 步以内使所有的灯都变亮。

输入格式

第一行输入正整数 n，代表数据中共有 n 个待解决的游戏初始状态。

以下若干行数据分为 n 组，每组数据有 5 行，每行 5 个字符。

每组数据描述了一个游戏的初始状态。

各组数据间用一个空行分隔。

输出格式

一共输出 nn 行数据，每行有一个小于等于 66 的整数，它表示对于输入数据中对应的游戏状态最少需要几步才能使所有灯变亮。

对于某一个游戏初始状态，若 66 步以内无法使所有灯变亮，则输出 −1−1。

数据范围

0<n≤500

解法

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

int n;
char s[10][10];
int dx[]={0,-1,0,1,0},dy[]={0,0,1,0,-1};
int INF = 10000;

void turn(int i, int j){
    for(int k=0;k<5;k++){
        int x = i + dx[k], y = j + dy[k];
        if(0<=x && x<=4 && 0<=y && y<=4){
        s[x][y] = '0' + ('1' - s[x][y]);
        }
    }
}

int work(){
    char backup[10][10];
    memcpy(backup,s,sizeof s);
    int ans = INF;
    for(int k=0;k<(1<<5);k++){
        int res = 0;
        for(int i=0;i<5;i++){
            if(k>>i & 1){
                turn(0,i);
                res++;
            }
        }
        for(int i=0;i<4;i++){
            for(int j=0;j<5;j++){
                if(s[i][j]=='0'){
                    turn(i+1,j);
                    res++;
                }
            }
        }
        bool is_successful = true;
        for(int i=0;i<5;i++){
            if(s[4][i]=='0'){
                is_successful = false;
                break;
            }
        }
        if(is_successful) ans = min(ans,res);
        memcpy(s,backup,sizeof s);
    }
    return ans>6?-1:ans;
}

int main()
{
    cin >> n;
    while(n--){
        for(int i=0;i<5;i++){
            cin >> s[i];
        }
        cout << work() << endl;
    }
    
    return 0;
}