[Codevs] 1014 棋盘染色

1049 棋盘染色

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 黄金 Gold

 

题目描述 Description

有一个5×5的棋盘，上面有一些格子被染成了黑色，其他的格子都是白色，你的任务的对棋盘一些格子进行染色，使得所有的黑色格子能连成一块，并且你染色的格子数目要最少。读入一个初始棋盘的状态，输出最少需要对多少个格子进行染色，才能使得所有的黑色格子都连成一块。（注：连接是指上下左右四个方向，如果两个黑色格子只共有一个点，那么不算连接）

输入描述 Input Description

   输入包括一个5×5的01矩阵，中间无空格，1表示格子已经被染成黑色。

输出描述 Output Description

输出最少需要对多少个格子进行染色

 

样例输入 Sample Input

11100

11000

10000

01111

11111

样例输出 Sample Output

1

分析 Analysis

思维级模版题= =。

判重：HASH蛤习

判断输出：DFS求联通块

主体：BFS

注意点：状态表示

主要还是数据范围小= =感谢出题人放我蒟蒻一马Orz

代码 Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
#define LL long long 
using namespace std;

const int dir[4][2] = {{0,1},{1,0},{-1,0},{0,-1}};

bool swi = true;

struct MAP{
    int map[7][7],step,remain;
    
    MAP(){
        memset(map,0,sizeof(map));
        step = remain = 0;
    }
};

queue<MAP> q;

bool HASH[21473650];

LL GETHASH(MAP now){
    LL code = 0;
    for(int i = 1;i <= 5;i++){
        for(int j = 1;j <= 5;j++){
            code = (code<<1)+now.map[i][j];
        }
    }
    
    return code%21473648;
}

bool poicheck(int x,int y){
    if(x > 5 || x < 1 || y > 5 || y < 1) return false;
    else return true;
}

int tot = 0;
bool chess[6][6];
void dfs(int x,int y,MAP now){
    if(chess[x][y] || !now.map[x][y] || !poicheck(x,y)) return;
    chess[x][y] = true;
    tot++;
    for(int i = 0;i < 4;i++){
        int nowx = x+dir[i][0];
        int nowy = y+dir[i][1];
        dfs(nowx,nowy,now);
    }
}

bool check(MAP now){
    int x,y;
    for(int i = 1;i <= 5;i++){
        for(int j = 1;j <= 5;j++){
            if(now.map[i][j]){
                x = i,y = j;
                break;
            }
        }
    }
    
    memset(chess,false,sizeof(chess));
    tot = 0;
    dfs(x,y,now);
    
    if(tot == 25-now.remain) return true;
    else return false;
}

void Input(){
    MAP tmp;
    
    for(int i = 1;i <= 5;i++){
        for(int j = 1;j <= 5;j++){
            scanf("%1d",&tmp.map[i][j]);
            if(!tmp.map[i][j]) tmp.remain++;
        } 
    }
    
    tmp.step = 0;
    
    
    
    q.push(tmp);
}

void PUSH(MAP now){
    now.step++;now.remain--;
    for(int i = 1;i <= 5;i++){
        for(int j = 1;j <= 5;j++){
            if(!now.map[i][j]){
                now.map[i][j] = 1;
                
                LL code = GETHASH(now);
                
                if(HASH[code]){
                    now.map[i][j] = 0;
                    continue;
                }else HASH[code] = 1;
                
                if(check(now)){
                    printf("%d",now.step);
                    swi = false;
                    return;
                }
                q.push(now);
                now.map[i][j] = 0;
            }
        }
    }
}

void bfs(){
    while(!q.empty() && swi){
        MAP tmp = q.front();
        q.pop();
//        cout << 'A';
        if(check(tmp)){
            printf("%d",tmp.step);
            swi = false;
            return;
        }
        
        PUSH(tmp);
    }
}

int main(){
    
    Input();
    bfs();
    
    return 0;
}