UVA1601 状态搜索

时间:2022-08-28 15:06:48

很有意思的一道题,就是迷宫问题的增强版。但是细节很多,有一个技巧就是把每个可以走的位置编号方便状态判重。

AC代码:

#include<cstdio>
#include<cstring>
#include<queue>
#include<iostream>
using namespace std;
const int maxs=20;
const int maxn=150;
const int dx[]={1,-1, 0, 0, 0};
const int dy[]={0, 0, 1,-1, 0};
int s[3],t[3];
int deg[maxn], G[maxn][5];
int d[maxn][maxn][maxn]; //记录步数
struct node //记录状态
{
    int a,b,c;
    node(int x,int y,int z):a(x),b(y),c(z){}
};

bool conflict(int a,int b,int a1,int b1)
{
    return a1==b1||(a1==b&&b1==a); //判断是否移动到同一个格子或者在一步中互换了位置
}

int bfs()
{
    queue<node>q;
    memset(d,-1,sizeof(d));
    d[s[0]][s[1]][s[2]]=0;
    q.push(node(s[0],s[1],s[2]));
    while(!q.empty())
    {
        node u=q.front(); q.pop();
        int a=u.a,b=u.b,c=u.c;
        if(a==t[0]&&b==t[1]&&c==t[2]) return d[a][b][c];
        //开始移动,移动时需要判断是否移动到同一个位置
        for(int i=0;i<deg[a];++i)
        {
            int a1=G[a][i];
            for(int j=0;j<deg[b];++j)
            {
                int b1=G[b][j];
                if(conflict(a,b,a1,b1)) continue;
                for(int k=0;k<deg[c];++k)
                {
                    int c1=G[c][k];
                    if(conflict(a,c,a1,c1)) continue;
                    if(conflict(b,c,b1,c1)) continue;
                    if(d[a1][b1][c1]!=-1)   continue;
                    d[a1][b1][c1]=d[a][b][c]+1;
                    q.push(node(a1,b1,c1));
                }
            }
        }
    }
}

int main()
{
    int w, h, n;
    while(scanf("%d%d%d\n",&w,&h,&n)==3&&w!=0)
    {
        char res[maxs][maxs];
        for(int i=0;i<h;++i) fgets(res[i],20,stdin);

        int cnt=0;
        int x[maxn],y[maxn],id[maxs][maxs];
        //给每个空格编号,很好的技巧,可以提高状态转换、判重的效率
        for(int i=0;i<h;++i)
        for(int j=0;j<w;++j)
        {
            if(res[i][j]!='#')
            {
                x[cnt]=i;y[cnt]=j;id[i][j]=cnt;
                if(islower(res[i][j])) s[res[i][j]-'a']=cnt;
                else if(isupper(res[i][j])) t[res[i][j]-'A']=cnt;
                ++cnt;
            }
        }

        for(int i=0;i<cnt;++i)
        {
            deg[i]=0;
            for(int dir=0;dir<5;++dir)
            {
                int px=x[i]+dx[dir],py=y[i]+dy[dir];
                if(px<0||px>=h||py<0&&py>=w) continue;
                if(res[px][py]!='#') G[i][deg[i]++]=id[px][py]; //当前空格的临近空格
            }
        }
        //为了简化代码,给少于三个鬼的情况增加到三个点,新增的点被处理为无法移动。
        if(n<=2) {deg[cnt] = 1; G[cnt][0] = cnt; s[2]=t[2]=cnt++;}
        if(n<=1) {deg[cnt] = 1; G[cnt][0] = cnt; s[1]=t[1]=cnt++;}

        printf("%d\n",bfs());
    }

    return 0;
}

如有不当之处欢迎指出!!