codevs 1049 棋盘染色

题目描述 Description

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

输入描述 Input Description

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

输出描述 Output Description

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

样例输入 Sample Input

11100

11000

10000

01111

11111

样例输出 Sample Output

1

well…搜索…先bfs找到黑的再扩展进行染色。判断么…用dfs判断是否连通。怀着高兴的心情提交上去。。。果然TLE了。。

呃呃好吧。我改一下判重。

原来是一个一个对比的，现在我把它转化成25位的二进制数，可以省去不少时间。

AC。。

#include <iostream>
#include <cstring>
using namespace std;

typedef char zt[][];
zt m,q[],v;
int f,r,num[],dx[]={-,,,},dy[]={,,-,};
bool vis[];
int black[][],count=-;
int s;

void dfs(zt ttpp,int xx,int yy)
{
    if(xx<||xx>=||yy<||yy>=||v[xx][yy]||ttpp[xx][yy]=='')
        return;
    v[xx][yy]=;
    for (int dd=;dd<;dd++)
        dfs(ttpp,xx+dx[dd],yy+dy[dd]);
}

int okay(zt tp)
{
    int i;
    memset(v,,sizeof(v));
    for (i=;i<;i++)
        if (tp[i/][i%]=='')
            break;
    dfs(tp,i/,i%);
    for (int i=;i<;i++)
        for (int j=;j<;j++)
            if (tp[i][j]=='' && !v[i][j])
                return ;
    return ;
}

int change_to_int(zt tt)
{
    int rrr=,d=;
    for (int i=;i>=;i--)
        for (int j=;j>=;j--)
        {
            rrr+=(tt[i][j]-'')*d;
            d*=;
        }
    return rrr;
}

int bfs()
{
    f=;r=;
    for (int i=;i<;i++)
        for (int j=;j<;j++)
            q[][i][j]=m[i][j];
    num[]=;
    s=change_to_int(m);
    vis[s]=;
    while (f<r)
    {
        f++;
        if (okay(q[f]))
            return num[f];
        count=-;
        for (int i=;i<;i++)
            for (int j=;j<;j++)
                if (q[f][i][j]=='')
                {
                    black[++count][]=i;
                    black[count][]=j;
                }
        for (int i=;i<=count;i++)
            for (int j=;j<;j++)
            {
                int nx=black[i][]+dx[j];
                int ny=black[i][]+dy[j];
                if (nx<||nx>=||ny<||ny>=||q[f][nx][ny]=='')
                    continue;
                r++;
                for (int i=;i<;i++)
                    for (int j=;j<;j++)
                    {
                        q[r][i][j]=q[f][i][j];
                    }
                q[r][nx][ny]='';
                s=change_to_int(q[r]);
                if (!vis[s])
                {
                    vis[s]=;
                    num[r]=num[f]+;
                }
                else
                    r--;
            }
    }
    return ;
}

int main()
{
    for (int i=;i<;i++)
        for (int j=;j<;j++)
            cin >> m[i][j];
    cout << bfs() << "\n";
}