【NOIP2013模拟】水叮当的舞步

【NOIP2013模拟】水叮当的舞步

题目

Description

水叮当得到了一块五颜六色的格子形地毯作为生日礼物，更加特别的是，地毯上格子的颜色还能随着踩踏而改变。
为了讨好她的偶像虹猫，水叮当决定在地毯上跳一支轻盈的舞来卖萌~~~
地毯上的格子有N行N列，每个格子用一个0~5之间的数字代表它的颜色。
水叮当可以随意选择一个0~5之间的颜色，然后轻轻地跳动一步，地毯左上角的格子所在的联通块里的所有格子就会变成她选择的那种颜色。这里连通定义为：两个格子有公共边，并且颜色相同。
由于水叮当是施展轻功来跳舞的，为了不消耗过多的真气，她想知道最少要多少步才能把所有格子的颜色变成一样的。

Input

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

Output

对于每组数据，输出一个整数，表示最少步数。

Sample Input

2
0 0
0 0
3
0 1 2
1 1 2
2 2 1
0

Sample Output

0
3

Data Constraint

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

题解

类型：IDA* (迭代加深启发式搜索)
明显的暴力题，但要加许多优化。
1、枚举答案，判断是否可行，可行即可输出。
2、每次只搜索左上角连通块旁边的颜色。
3、设置估价函数。每搜索到一个状态，若 非连通块的颜色总数+现在步数>ans就可以return。
4、发现每次找连通块耗时巨大，我们引入一个n*n的v数组。左上角的格子所在的联通块里的格子标记为1，左上角联通块周围一圈格子标记为2，其它格子标记为0。如果某次选择了颜色c，我们只需要找出标记为2并且颜色为c的格子，向四周扩展，并相应地修改v标记，就可以不断扩大标记为1的区域，最终如果所有格子标记都是1，那么显然找到了答案。

#include<cmath>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
using namespace std;
int n,ans,a[10][10],fx[4][2]={{1,0},{-1,0},{0,1},{0,-1}},v[10][10];
bool bz[10],f;
void cg1(int x,int y)
{
    for (int k=0;k<=3;k++)
    {
        int x1=x+fx[k][0],y1=y+fx[k][1];
        if (x1>0 && x1<=n && y1>0 && y1<=n && v[x1][y1]==0) 
        v[x1][y1]=2;
    }
}
void change(int x,int y)
{
    for (int i=0;i<=3;i++)
    {
        int x1=x+fx[i][0],y1=y+fx[i][1];
        if (x1>0 && x1<=n && y1>0 && y1<=n && v[x1][y1]==2 && a[x1][y1]==a[x][y])   
        {
            v[x1][y1]=1;
            cg1(x1,y1);
            change(x1,y1);
        }
    }   
}
void gd(int x)
{
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        if (v[i][j]==2 && a[i][j]==x)
        {
            v[i][j]=1;
            cg1(i,j);   
            change(i,j);
        }
}
void dg(int x)
{
    int m=0;
    memset(bz,false,sizeof(bz));
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        if (v[i][j]!=1 && !bz[a[i][j]])
        {
            m++;
            bz[a[i][j]]=true;
        }
    if (x+m>ans) return;
    if (m==0)
    {
        f=true;
        return; 
    } 
    int b[10],l=0;
    memset(bz,false,sizeof(bz));
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        if (v[i][j]==2 && !bz[a[i][j]])
        {
            bz[a[i][j]]=true;
            b[++l]=a[i][j];
        }
    int v1[10][10];
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++) v1[i][j]=v[i][j];
    for (int i=1;i<=l;i++)
    {
        gd(b[i]);
        dg(x+1);
        if (f) return;
        for (int j=1;j<=n;j++)
            for (int k=1;k<=n;k++) v[j][k]=v1[j][k];
    }
}
int main()
{
    scanf("%d\n",&n);
    while (n!=0)
    {
        for (int i=1;i<=n;i++)
            for (int j=1;j<=n;j++) scanf("%d",&a[i][j]);
        memset(v,0,sizeof(v));
        v[1][1]=1;
        v[1][2]=2;
        v[2][1]=2;
        gd(a[1][1]);
        for (int i=0;;i++)
        {
            ans=i;
            f=false;
            dg(0);
            if (f) break;
        }
        printf("%d\n",ans);
        scanf("%d\n",&n);   
    }
    return 0;
}