CodeVS2495 水叮当的舞步

标签：IDA*

题目

题目传送门

题目描述 Description

　　水叮当得到了一块五颜六色的格子形地毯作为生日礼物，更加特别的是，地毯上格子的颜色还能随着踩踏而改变。
　　为了讨好她的偶像虹猫，水叮当决定在地毯上跳一支轻盈的舞来卖萌~~~

　　地毯上的格子有N行N列，每个格子用一个0~5之间的数字代表它的颜色。
　　水叮当可以随意选择一个0~5之间的颜色，然后轻轻地跳动一步，左上角的格子所在的联通块里的所有格子就会变成她选择的那种颜色。这里连通定义为：两个格子有公共边，并且颜色相同。
　　由于水叮当是施展轻功来跳舞的，为了不消耗过多的真气，她想知道最少要多少步才能把所有格子的颜色变成一样的。
输入描述 Input Description

　　每个测试点包含多组数据。
　　每组数据的第一行是一个整数N，表示地摊上的格子有N行N列。
　　接下来一个N*N的矩阵，矩阵中的每个数都在0~5之间，描述了每个格子的颜色。
　　N=0代表输入的结束。
输出描述 Output Description

　　对于每组数据，输出一个整数，表示最少步数。
样例输入 Sample Input

2
0 0
0 0
3
0 1 2
1 1 2
2 2 1
0
样例输出 Sample Output

0
3
数据范围及提示 Data Size & Hint

　　对于30%的数据，N<=5
　　对于50%的数据，N<=6
　　对于70%的数据，N<=7
　　对于100%的数据，N<=8，每个测试点不多于20组数据。

第二组样例解释：
　　0 1 2 1 1 2 2 2 2 1 1 1
　　1 1 2 –> 1 1 2 –> 2 2 2 –> 1 1 1
　　2 2 1 2 2 1 2 2 1 1 1 1

来源：Nescafe 21

分析

黄学长题解

我们引入一个N*N的vis数组。左上角的格子所在的联通块里的格子标记为1。左上角联通块周围一圈格子标记为2，其它格子标记为0。如果某次选择了颜色c，我们只需要找出标记为2并且颜色为c的格子，向四周扩展，并相应地修改v标记，就可以不断扩大标记为1的区域，最终如果所有格子标记都是1，那么显然找到了答案。

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=last[x];i;i=e[i].next)
using namespace std;
inline ll read()
{
    ll f=1,x=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int dx[4]={0,1,0,-1},dy[4]={1,0,-1,0};
int Map[9][9],vis[9][9],n,lim,used[6];
bool ans;
int get(){
    int t=0;mem(used,0);
    rep(i,1,n)
        rep(j,1,n)
            if(!used[Map[i][j]]&&vis[i][j]!=1){
                used[Map[i][j]]=1;
                t++;
            }
    return t;
}
void dfs(int a,int b,int x){
    vis[a][b]=1;
    rep(i,0,3){
        int nx=a+dx[i],ny=b+dy[i];
        if(nx<1||nx>n||ny<1||ny>n||vis[nx][ny]==1)continue;
        vis[nx][ny]=2;
        if(Map[nx][ny]==x)dfs(nx,ny,x);
    }
}

int fill(int x){
    int t=0;
    rep(i,1,n)
        rep(j,1,n)
            if(vis[i][j]==2&&Map[i][j]==x){
                t++;
                dfs(i,j,x);
            }
    return t;
}
void search(int k){
    int v=get();
    if(!v)ans=1;
    if(k+v>lim||ans)return;
    int tmp[10][10];
    rep(i,0,5){
        memcpy(tmp,vis,sizeof vis);
        if(fill(i))search(k+1);
        memcpy(vis,tmp,sizeof vis);
    }
}

int main()
{
    while(true){
        mem(vis,0);ans=0;lim=-1;
        n=read();if(n==0)break;
        rep(i,1,n)
            rep(j,1,n)Map[i][j]=read();
        dfs(1,1,Map[1][1]);
        while(true){
            lim++;
            search(0);
            if(ans){printf("%d\n",lim);break;}
        }
    }
    return 0;
}