[HDU 1254] 推箱子

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1254

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

#define INF 0xfffffff

const int maxn = 10;
int go[4][2] = {0,1,0,-1,1,0,-1,0};
int maze[10][10];
int vis[10][10][10][10]; //是由人和箱子的位置这两个状态共同决定
int m,n;

struct Point
{
    int x,y;
    int bx,by;
    int cnt;
}P;

//用这个函数替代memset(),不然会导致wronganswer
void init()
{
    for(int i=0; i<maxn; i++)
        for(int j=0; j<maxn; j++)
            for(int k=0; k<maxn; k++)
                for(int p=0; p<maxn; p++)
                    vis[i][j][k][p] = INF;
}

bool OKman(Point s)
{
    return s.x>=0 && s.x<m && s.y>=0 && s.y<n && maze[s.x][s.y]!=1 && s.cnt < vis[s.x][s.y][s.bx][s.by];
}

bool OKbox(Point s)
{
    return s.bx>=0 && s.bx<m && s.by>=0 && s.by<n && maze[s.bx][s.by]!=1 && s.cnt < vis[s.x][s.y][s.bx][s.by];
}

int BFS()
{
    Point Pn;
    queue<Point> Q;
    Q.push(P);
    int ans = INF;
    vis[P.x][P.y][P.bx][P.by] = 1;
    while(!Q.empty())
    {
        P = Q.front();
        Q.pop();
        if(maze[P.bx][P.by] == 3)
        {
            ans = min(ans,P.cnt);
        }
        for(int i=0;i<4;i++)
        {
            Pn = P;
            Pn.x += go[i][0];
            Pn.y += go[i][1];
            if(OKman(Pn)) //判断人走下一步是否合法
            {
                if(Pn.x == Pn.bx && Pn.y == Pn.by) //在人走下一步合法的条件下,判断人和箱子是否重合,重合就得移动箱子
                {
                    Pn.bx += go[i][0];
                    Pn.by += go[i][1];
                    Pn.cnt++;
                    if(OKbox(Pn)) //判断箱子走的下一步是否合法,合法就加入队列
                    {
                        vis[Pn.x][Pn.y][Pn.bx][Pn.by] = Pn.cnt;
                        Q.push(Pn);
                    }
                }
                else //如果人和箱子不重合,也就是人还不能推箱子
                {
                    vis[Pn.x][Pn.y][Pn.bx][Pn.by] = Pn.cnt;
                    Q.push(Pn);
                }
            }
        }
    }
    return ans;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d%d",&m,&n);
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
            {
                scanf("%d",&maze[i][j]);
                if(maze[i][j] == 4)
                    P.x=i,P.y=j,P.cnt=0;
                else if(maze[i][j] == 2)
                    P.bx=i,P.by=j;
            }
//        memset(vis,INF,sizeof(vis));  用memset会导致wronganswer,原因后续跟上.
        int k = BFS();
        if(k == INF)
            k = -1;
        printf("%d
",k);
    }
    return 0;
}