PTA 画家问题 bitset 易懂

有一个正方形的墙，由NN个正方形的砖组成，其中一些砖是白色的，另外一些砖是黄色的。Bob是个画家，想把全部的砖都涂成黄色。但他的画笔不好使。当他用画笔涂画第(i,j)个位置的砖时， 位置(i−1,j)、 (i+1,j)、(i,j−1)、(i,j+1)上的砖都会改变颜色。请你帮助Bob计算出最少需要涂画多少块砖，才能使所有砖的颜色都变成黄色。

输入格式:

第一行是一个整数n (1≤n ≤16)，表示墙的大小。接下来的n行表示墙的初始状态。每一行包含n个字符。第i行的第j个字符表示位于位置(i,j)上的砖的颜色。“w”表示白砖，“y”表示黄砖。

输出格式:

一行，如果Bob能够将所有的砖都涂成黄色，则输出最少需要涂画的砖数，否则输出“inf”。

 很典型的熄灯问题，通过bitset维护状态暴力求解。

#include<bits/stdc++.h>
using namespace std;
//待到秋来九月八，我花开后百花杀
int n;
bitset<17>mp[17],raw[17];
// Memrony: 488KB Time:108ms
signed main(){
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++){
        for(int j=0;j<n;j++){
            char ch;cin>>ch;
            mp[i][j] = ch=='w' ? 1 : 0;
        }
        raw[i] = mp[i];//存原图
    }
    bool flag = false;
    int ans = -1;
    for(int k=0;k<(1<<n);k++){
        mp[0] = k;
        int cnt = 0;
        for(int i=1;i<=n;i++){
            mp[i] = raw[i];// 还原地图
        }
        // 尝试修改状态
        for(int i=1;i<=n;i++){
            for(int j=0;j<n;j++){
                if(mp[i-1][j]){
                    cnt++;
                    mp[i][j] = !mp[i][j];
                    if(j>0)mp[i][j-1] = !mp[i][j-1];
                    if(j<n-1)mp[i][j+1] = !mp[i][j+1];
                    mp[i+1][j] = !mp[i+1][j];
                }
            }
        }
        if(mp[n] == 0){// 全部完成，取最小操作次数
            flag = true;
            if(ans == -1 || ans > cnt)ans = cnt;
        }


    }
    if(flag)cout<<ans<<endl;
    else cout<<"inf"<<endl;


    return 0;
}

 通过截图：