PTA 2813：画家问题（熄灯问题）

最新推荐文章于 2024-08-24 14:08:21 发布

HKer_YM

最新推荐文章于 2024-08-24 14:08:21 发布

阅读量262

点赞数 18

文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/HKer_YM/article/details/137614167

版权

PTA 2813：画家问题（熄灯问题）

经典的熄灯问题，思路就不过多阐述，csdn都有

#include <bits/stdc++.h>
using namespace std;
int num[20];
int num2[20];
int main() {

    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        string str;
        cin >> str;
        for (int j = 0; j < n; j++) {
            // 如果是开灯状态，标记上1，反之0
            num[i] |= str[j] == 'w' ? 1 << j : 0;
        }
        // 存原图
        num2[i] = num[i];
    }
    // 结果可不可行
    bool flag = false;
    // 最小操作次数
    int ans = -1;
    // 2 ==> 00, 01, 10, 11,
    // 2  0 1 2 3
    // 0 ~ 2^2 - 1 (1 << 2) -1
    // 枚举N个格子的状态
    for (int k = 0; k < (1 << n); k++) {
        // 第0层的状态为枚举的K
        num[0] = k;
        // 这次枚举的开关灯次数
        int cnt = 0;
        // 还原图，因为上次实在num上修改的，所以要还原
        for (int i = 1; i <= n; i++) {
            num[i] = num2[i];
        }
        // 修改灯的状态
        for (int i = 1; i <= n; i++) {
            for (int j = 0; j < n; j++) {
                // 01 & 10 == 00 == 0
                // 如果上一层的第j个灯是开的，则这一层需要关闭
                if (num[i - 1] & (1 << j)) {
                    // 次数++
                    cnt++;
                    // 修改当前灯的状态
                    num[i] ^= (1 << j);
                    // 左边灯的状态
                    if (j > 0) num[i] ^= (1 << (j - 1));
                    // 右边灯的状态
                    if (j < n - 1) num[i] ^= (1 << (j + 1));
                    // 下一层第j个灯的状态
                    num[i + 1] ^= (1 << j);
                    // 上一层第j个灯不用修改，因为最后我们只需要判断最后一层灯的状态
                }
            }
        }
        // 如果最后一层灯都关闭了，则二进制上都是0，所以结果为0
        if (num[n] == 0) {
            flag = true;
            // 更新答案
            if (ans == -1 || ans > cnt) ans = cnt;
        }
    }
    if(!flag) cout << "inf" << endl;
    else cout << ans << endl;
    return 0;
}