POJ 1847 Tram --set实现最短路SPFA

题意很好懂，但是不好下手。这里可以把每个点编个号（1-25），看做一个点，然后能够到达即为其两个点的编号之间有边，形成一幅图，然后求最短路的问题。并且pre数组记录前驱节点，print_path()方法可用算法导论上的。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#define Mod 1000000007
#define INT 2147483647
#define pi acos(-1.0)
#define eps 1e-3
#define lll __int64
#define ll long long
using namespace std;
#define N 100007

const int INF = Mod;
vector<pair<int,int> > edge
;
set<pair<int,int> > que;
int a[6][6],n;
int d
,pre
;

void print_path(int s,int v)
{
    if(v == s)
        printf("(%d, %d)\n",(s-1)/5,(s%5+5-1)%5);
    else
    {
        print_path(s,pre[v]);
        printf("(%d, %d)\n",(v-1)/5,(v%5+5-1)%5);
    }
}

int ok(int x,int y)
{
    if(x >= 0 && x < 5 && y >= 0 && y < 5)
        return 1;
    return 0;
}

void SPFA()
{
    int i;
    for(i=2;i<=n;i++)
        d[i] = INF;
    d[1] = 0;
    que.insert(make_pair(d[1],1));
    while(!que.empty())
    {
        int v = que.begin()->second;
        que.erase(que.begin());
        for(i=0;i<edge[v].size();i++)
        {
            int to = edge[v][i].first;
            int cost = edge[v][i].second;
            if(d[v] + cost < d[to])
            {
                que.erase(make_pair(d[to],to));
                d[to] = d[v] + cost;
                pre[to] = v;
                que.insert(make_pair(d[to],to));
            }
        }
    }
    print_path(1,n);
}

int main()
{
    int i,j;
    for(i=0;i<5;i++)
        for(j=0;j<5;j++)
            scanf("%d",&a[i][j]);
    for(i=0;i<5;i++)
    {
        for(j=0;j<5;j++)
        {
            if(a[i][j] == 0)
            {
                int ka = i*5 + j + 1,kb;
                if(ok(i+1,j) && a[i+1][j] == 0)
                {
                    kb = ka + 5;
                    edge[ka].push_back(make_pair(kb,1));
                }
                if(ok(i,j+1) && a[i][j+1] == 0)
                {
                    kb = ka+1;
                    edge[ka].push_back(make_pair(kb,1));
                }
                if(ok(i-1,j) && a[i-1][j] == 0)
                {
                    kb = ka-5;
                    edge[ka].push_back(make_pair(kb,1));
                }
                if(ok(i,j-1) && a[i][j-1] == 0)
                {
                    kb = ka-1;
                    edge[ka].push_back(make_pair(kb,1));
                }
            }
        }
    }
    n = 25;
    SPFA();
    return 0;
}

View Code