【算法】费解的开关(递推)

题目描述 

你玩过“拉灯”游戏吗?

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

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

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

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

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

下面这种状态

10111
01101
10111
10000
11011

在改变了最左上角的灯的状态后将变成:

01111
11101
10111
10000
11011

再改变它正中间的灯后状态将变成:

01111
11001
11001
10100
11011

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

输入格式

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

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

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

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

输出格式

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

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

数据范围

0 < n ≤ 500

输入样例:

3
00111
01011
10001
11010
11100

11101
11101
11110
11111
11111

01111
11111
11111
11111
11111

输出样例:

3
2
-1

思路

        仔细观察可以发现,当第一排的灯状态确定之后,第2,3,4,5排的灯状态也就确定了。

第一步:搜索第一排灯开关状态改变或不改变(一共有32种结果)

第二步:根据第一排的灯的状态判断第二排的灯的开关是否需要改变状态(若前一排的灯为0,则改变它下面的那盏灯开关的状态),以此类推到第五排。

第三步:判断最后一排的灯是否全亮,若全部为亮,则比较开关改变的次数。若不全为亮,则进行下一次循环(进入第一步)。

代码 

#include
using namespace std;
char str[6][6];
char cpy[6][6];
int dx[5] = {1,-1,0,0,0};
int dy[5] = {0,0,-1,1,0};
void turn(int x,int y)
{
    for(int i = 0; i < 5; i ++)
    {
        int xx = x + dx[i];
        int yy = y + dy[i];
        if(xx < 0 || xx > 4 || yy < 0 || yy > 4) continue;
        if(str[xx][yy] == '1') str[xx][yy] = '0';
        else str[xx][yy] = '1';
    }
}

void solve()
{
    for(int i = 0; i < 5; i ++) cin >> str[i];
    memcpy(cpy,str,sizeof str);// 对数组进行操作后需要进行复原
    int res = 10;
    for(int op = 0; op < 32; op ++)
    {
        int cnt = 0;
        memcpy(str,cpy,sizeof str);
        // 对第一行进行操作
        for(int i = 0; i < 5; i ++)
        {
            if(op >> i & 1)
            {
                turn(0,i);
                cnt ++;
            }
        }
        // 对剩下四行进行操作
        for(int i = 0; i < 4; i ++)
        {
            for(int j = 0; j < 5; j ++)
            {
                if(str[i][j] == '0')
                {
                    turn(i + 1,j);
                    cnt ++;
                }
            }
        }
        bool success = true;
        // 判断最后一行是否全亮
        for(int i = 0; i < 5; i ++)
        {
            if(str[4][i] == '0') success = false;
        }
        if(success && cnt < res) res = cnt;
    }
    if(res > 6) res = -1;
    cout << res << endl;
}


int main()
{
    int t;
    cin >> t;
    while(t --)
        solve();
    return 0;
}

你可能感兴趣的:(算法,算法,数据结构,递推,Acwing)