hdu 1732 Push Box

http://acm.hdu.edu.cn/showproblem.php?pid=1732

推箱子和游戏规则一样。

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;

char g[10][10];
int n,m;
int sx,sy;
bool vis[9][9][9][9][9][9][9][9];
int dir[4][2]= {{0,1},{0,-1},{1,0},{-1,0}};
struct node
{
    int x[4],y[4],xx,yy,step;
} st1,st2,st;

int deal(node p,int i,int pos)
{
    p.xx+=dir[i][0];
    p.yy+=dir[i][1];
    if(p.xx>=0&&p.xx<n&&p.yy>=0&&p.yy<m)
    {
        for(int j=0; j<3; j++)
        {
            if(j!=pos&&p.x[j]==p.xx&&p.y[j]==p.yy)
            {
                return 0;
            }
        }
        return 1;
    }
    return 0;
}

int bfs()
{
    queue<node>q;
    st.step=0;
    st.xx=sx;
    st.yy=sy;
    q.push(st);
    memset(vis,false,sizeof(vis));
    vis[sx][sy][st.x[0]][st.y[0]][st.x[1]][st.y[1]][st.x[2]][st.y[2]]=true;
    while(!q.empty())
    {
        st1=q.front();
        q.pop();
        int cnt=0;
        for(int i=0; i<3; i++)
        {
            if(g[st1.x[i]][st1.y[i]]=='@')
            {
                cnt++;
            }
        }
        if(cnt==3)
        {
            return st1.step;
        }
        for(int i=0; i<4; i++)
        {
            st2=st1;
            st2.xx=st2.xx+dir[i][0];
            st2.yy=st2.yy+dir[i][1];
            st2.step++;
            if(st2.xx>=0&&st2.xx<n&&st2.yy>=0&&st2.yy<m&&g[st2.xx][st2.yy]!='#')
            {
                int pos;
                for(pos=0; pos<3; pos++)
                {
                    if(st2.x[pos]==st2.xx&&st2.y[pos]==st2.yy)
                    {
                        break;
                    }
                }
                if(pos<3)
                {
                    if(deal(st2,i,pos))
                    {
                        st2.x[pos]+=dir[i][0];
                        st2.y[pos]+=dir[i][1];
                        if(!vis[st2.xx][st2.yy][st2.x[0]][st2.y[0]][st2.x[1]][st2.y[1]][st2.x[2]][st2.y[2]])
                        {
                            vis[st2.xx][st2.yy][st2.x[0]][st2.y[0]][st2.x[1]][st2.y[1]][st2.x[2]][st2.y[2]]=true;
                            q.push(st2);
                        }
                    }
                }
                else
                {
                    if(!vis[st2.xx][st2.yy][st2.x[0]][st2.y[0]][st2.x[1]][st2.y[1]][st2.x[2]][st2.y[2]])
                    {
                        vis[st2.xx][st2.yy][st2.x[0]][st2.y[0]][st2.x[1]][st2.y[1]][st2.x[2]][st2.y[2]]=true;
                        q.push(st2);
                    }
                }
            }
        }
    }
    return -1;
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        int num=0;
        for(int i=0; i<n; i++)
        {
            scanf("%s",g[i]);
            for(int j=0; j<m; j++)
            {
                if(g[i][j]=='X')
                {
                    g[i][j]='.';
                    sx=i;
                    sy=j;
                }
                else if(g[i][j]=='*')
                {
                    g[i][j]='.';
                    st.x[num]=i;
                    st.y[num]=j;
                    num++;
                }
            }
        }
        printf("%d
",bfs());
    }
    return 0;
}