【算法学习笔记】74. 枚举 状态压缩 填充方案 SJTU OJ 1391 畅畅的牙签袋（改）

#include <iostream>
#include <algorithm>
using namespace std;
const int INF = 9999999;
bool map[20][20]={0};
bool cur[20][20]={0};
int H,W;

void print(){
    cout<<"--------"<<endl;
    for (int i = 0; i < H; ++i){
        for (int j=0; j < W; ++j){
            cout<<cur[i][j];
        }
        cout<<endl;
    }
    cout<<"--------"<<endl;
}

void copyMap(){
    for (int i = 0; i < H; ++i)
        for (int j=0; j < W; ++j)
            cur[i][j] = map[i][j];
}


void init(){
    cin>>H>>W;
    for (int i = 0; i < H; ++i){
        for (int j = 0; j < W; ++j){
            int l;
            cin>>l;
            map[i][j] = l%2; //偶数为0 奇数为1
        }
    }
}

//计算k的二进制中有多少个1
int getOnes(int k){
    int ans = 0;
    while(k>0){
        if((k&1)) //如果最右侧是1
            ans++;
        k = k>>1;//每次扔出最右的一位;
    }
    return ans;
}

void put(int lineId, int state){
    //枚举时我们认为一个 状态数state 每个位置的 0表示不放 1表示放 那么进行填补
    for (int i = 0; i < W ; ++i) //从第一排第0个位置 到第W-1个
    {
        if(state&(1<<i)){//表示是放的 则对两边及下方的进行填写
            if(lineId>=1)//可以不用写
                cur[lineId-1][i] = !cur[lineId-1][i];
            cur[lineId][i] = !cur[lineId][i];
            if(i>=1)
                cur[lineId][i-1] = !cur[lineId][i-1];
            cur[lineId][i+1] = !cur[lineId][i+1];
            cur[lineId+1][i] = !cur[lineId+1][i];
        }
    }
}


int build(){
    
    bool have = false;
    int res = INF;
    //首先要对第一排的放十字的所有可能情况进行枚举
    //然后根据第一行的摆放情况 可以确定第二行的摆放,依次决定第三行.....
    for (int i = 0; i < (1<<W); ++i)//一共有2^W种放的情况
    {
        //每次枚举前先进行copy 不能直接在map上进行修改
        copyMap();
        int cnt = getOnes(i);//对放的十字的个数进行记录
        put(0,i);
        //print();
        //第一排做完了 我们来根据第一排的情况 确定接下来每一排必须放十字的位置
        for (int k = 1; k < H; ++k)
        {
            int curState = 0;//生成状态数 用来进行调用put函数
            for (int j = 0; j < W; ++j)
            {
                if(cur[k-1][j])//如果上一行的这个地方是1 即是奇数 则肯定要在以这一行的这个位置为中心进行填补
                    curState += (1<<j);
            }
            cnt += getOnes(curState);
            put(k,curState);
            
        }
        //放完之后要检查最后一行是不是已经被填满 如果是的话 则说明全体都被填成偶数
        bool ok = true;
        for (int j = 0; j < W; ++j)
        {
            if(cur[H-1][j]){
                ok = false;
                break;
            }
        }
        if(ok){
            have = true;
            res = min(res,cnt);
        }
    }
    if(!have)
        res = -1;
    return res;
}


int main(int argc, char const *argv[])
{
    init();
    cout<<build()<<endl;
    return 0;
}