HDU1026 - Ignatius and the Princess I （广搜+优先队列）

题目链接

思路

这道题的思路其实是很简单的，就是一个广搜，找最 短 路，区别就是在每个方格待的时间是不同的，所以需要一个优先队列来保证每次访问的都是已花费时间最小的，而且在每次访问节点时，都要将当前节点的信息更新为最小时间花费，还要记录上一个节点的坐标，最后利用栈来调整输出顺序。
(PS:基本功太差，所以当我花费很长时间将这道题调试正确的时候，我的内心是崩溃的。)

代码

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <stack>

using namespace std;
const int inf = 0x7fffffff;

struct pos
{
    int x, y;
    pos(int _x, int _y):x(_x), y(_y) {}
};

struct node
{
    int time, cost;
    int prex, prey;
    bool can_walk = false;
} map[105][105];
bool vis[105][105];

struct cmp
{
    bool operator()(const pos &a, const pos &b)
    {
        return map[a.x][a.y].time > map[b.x][b.y].time;
    }
};

int main()
{
    int n, m;
    int curx, cury;
    char sta[110];

    while(scanf("%d%d", &n, &m)==2)
    {
        // 读入数据，为了操作方便，下面的操作相当于在迷宫外面套了一层墙
        for(int i=1; i<=n; i++)
        {
            scanf("%s", sta+1);
            for(int j=1; j<=m; j++)
            {
                if(sta[j]=='X')
                {
                    map[i][j].can_walk = false;
                }
                else if(sta[j]=='.')
                {
                    map[i][j].can_walk = true;
                    map[i][j].time = inf;
                    map[i][j].cost = 1;
                }
                else
                {
                    map[i][j].can_walk = true;
                    map[i][j].time = inf;
                    map[i][j].cost = sta[j] - '0' + 1;
                }
            }
            map[i][m+1].can_walk = false;
        }
        for(int i=1; i<=m; i++) map[n+1][i].can_walk = false;

        // 广搜
        memset(vis, false, sizeof(vis));
        priority_queue<pos, vector<pos>, cmp> qu;
        qu.push(pos(1, 1));
        map[1][1].time = 0;
        while(!qu.empty())
        {
            curx = qu.top().x;
            cury = qu.top().y;
//            printf("cur : %d %d\n", curx, cury);
            qu.pop();
            if(vis[curx][cury]) continue;


            // 看上去很多，其实四个方向的操作是一样的，以一个为例
            // 此节点可通行，并未访问
            if(map[curx][cury+1].can_walk && !vis[curx][cury+1])
            {
                // 进入队列
                qu.push(pos(curx, cury+1));
                // 更新信息，记录最小时间及前一节点
                if((map[curx][cury].time+map[curx][cury].cost)<map[curx][cury+1].time)
                {
                    map[curx][cury+1].time = map[curx][cury].time+map[curx][cury].cost;
                    map[curx][cury+1].prex = curx;
                    map[curx][cury+1].prey = cury;
                }
            }
            if(map[curx+1][cury].can_walk && !vis[curx+1][cury])
            {
                qu.push(pos(curx+1, cury));
                if((map[curx][cury].time+map[curx][cury].cost)<map[curx+1][cury].time)
                {
                    map[curx+1][cury].time = map[curx][cury].time+map[curx][cury].cost;
                    map[curx+1][cury].prex = curx;
                    map[curx+1][cury].prey = cury;
                }
            }
            if(map[curx][cury-1].can_walk && !vis[curx][cury-1])
            {
                qu.push(pos(curx, cury-1));
                if((map[curx][cury].time+map[curx][cury].cost)<map[curx][cury-1].time)
                {
                    map[curx][cury-1].time = map[curx][cury].time+map[curx][cury].cost;
                    map[curx][cury-1].prex = curx;
                    map[curx][cury-1].prey = cury;
                }
            }
            if(map[curx-1][cury].can_walk && !vis[curx-1][cury])
            {
                qu.push(pos(curx-1, cury));
                if((map[curx][cury].time+map[curx][cury].cost)<map[curx-1][cury].time)
                {
                    map[curx-1][cury].time = map[curx][cury].time+map[curx][cury].cost;
                    map[curx-1][cury].prex = curx;
                    map[curx-1][cury].prey = cury;
                }
            }

            // 标记为已访问
            vis[curx][cury] = true;
        }


        // 输出结果
        if(map[n][m].time==inf) printf("God please help our poor hero.\n");
        else
        {
            // 用栈来调整输出顺序
            stack<pos> st;
            pos temp(n, m);
            while(true)
            {
                st.push(temp);
//                printf("%d %d\n", temp.x, temp.y);
                if(temp.x==1 && temp.y==1) break;
                temp = pos(map[temp.x][temp.y].prex, map[temp.x][temp.y].prey);
            }

            printf("It takes %d seconds to reach the target position, let me show you the way.\n", map[n][m].cost+map[n][m].time-1);
            int curt = 1;
            while(!st.empty())
            {
                curx = st.top().x;
                cury = st.top().y;
                st.pop();

                for(int i=2; i<=map[curx][cury].cost; i++)
                    printf("%ds:FIGHT AT (%d,%d)\n", curt++, curx-1, cury-1);
                if(!st.empty())
                {
                    printf("%ds:(%d,%d)->(%d,%d)\n", curt++, curx-1, cury-1, st.top().x-1, st.top().y-1);
                }
            }
        }
        printf("FINISH\n");
    }
    return 0;
}