BFS uva1601 万圣节后的早晨

 

难点在于如何记录当前的状态，存入栈中又将其转化为位置信息。

#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
#include<cstdio>
using namespace std;
int w,h,n;
int d[256][256][256];
vector<int>g[256];
int up[3],low[3];
string map[17];
int x[5]={1,0,-1,0,0};
int y[5]={0,1,0,-1,0};
inline int getid(int a,int b,int c)//inline为内联函数，运行更快
{
    return (a<<16)|(b<<8)|c;
}
bool jy(int a,int b,int x,int y)
{
    return (a==y&&b==x)||x==y;
}
int bfs()
{
    if(low[0]==up[0]&&low[1]==up[1]&&low[2]==up[2])
    return 0;
    queue<int>q;
    memset(d,-1,sizeof(d));
    q.push(getid(low[0],low[1],low[2]));
    d[low[0]][low[1]][low[2]]=0;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        int a=(u>>16)&0xff;
        int b=(u>>8)&0xff;
        int c=u&0xff;
        for(int i=0;i<g[a].size();i++)
        {
            int x=g[a][i];
            for(int j=0;j<g[b].size();j++)
            {
                int y=g[b][j];
                if(jy(a,b,x,y))continue;
                for(int k=0;k<g[c].size();k++)
                {
                    int z=g[c][k];
                    if(jy(a,c,x,z)||jy(b,c,y,z)||d[x][y][z]!=-1)continue;
                    if(x==up[0]&&y==up[1]&&z==up[2]) return d[a][b][c]+1;
                     d[x][y][z]=d[a][b][c]+1;
                     q.push(getid(x,y,z));
                }
            }
        }
    }
    return -1;
}
int main()
{    
    while(cin>>w>>h>>n&&n){
    getline(cin,map[0]);
    for(int i=0;i<256;i++)
    g[i].clear();//每次录入要清理原先数据！！这里忘了写导致我改了好久
    for(int i=0;i<h;i++)
    getline(cin,map[i]);
    for(int i=0;i<h;i++)
    {
        for(int j=0;j<w;j++)
        {
            if(map[i][j]!='#')
            {
                for(int k=0;k<5;k++)//记录五种可能移动的位置（位置是0-255之间的一个数）
                {
                    int dx=i+x[k];
                    int dy=j+y[k];
                    char c=map[dx][dy];
                    if(c!='#')//可以移动
                    {
                        int a=i*w+j;
                        int b=dx*w+dy;
                        if(isupper(c))up[c-'A']=b;//记录大写位置
                        else if(islower(c))low[c-'a']=b;//记录小写位置
                        g[a].push_back(b);
                    }
                }
            }
        }
    }
        if(n<3){
            up[2]=low[2]=2;g[2].push_back(2);//这里可以随便写  0 1 2 3 都行 应为第一行必然都是‘#’
            if(n<2)up[1]=low[1]=1;g[1].push_back(1);
        }
        cout<<bfs()<<endl;
    }
    return 0;
}