hdu-4725 The Shortest Path in Nya Graph （spfa + 建图）

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

题目大意：在一个图中有n个点，m条带权无向边，同时这n个点是分布在n层当中的，任意相邻的两层的点可以互达，从一层到另一层所需要的花费为c，现在求从点1到点n的最短路径。


题目思路：本题就是一个最经典的最短路问题，但是难点在于建图。如何处理不同层的点之间的距离是本题的最大难点。我们可以将n层表示为n+1~2*n这n个点，相邻的层如果都存在点，那么相邻的两层之间就可以建立一条权值为c的边。如果点i是在第layer[i]层，那么就可以在点i和点n+layer[i]之间建立一条权值为0的边，在点i与点n+layer[i]-1和点n+layer[i]之间分别建立一条权值为c的边，这样就可以通过n+1~2*n这n个点来完成不同楼层点的移动。建完图后直接用个spfa跑一遍最短路即可求得结果。

AC代码如下：

#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
#define FIN freopen("in.txt","r",stdin)
#define fuck(x) cout<<'['<<x<<']'<<endl
using namespace std;
typedef long long LL;
typedef pair<int, int>pii;
const int MX = 2e5 + 10;//有额外的n个点来储存不同层的状态，所以要开大两倍；

struct edge {
    int v, w;
    int nxt;
} E[MX * 20];
int head[MX], tot;
int T, n, m, c;
int layer[MX], vv[MX];

void init() {
    memset(head, -1, sizeof(head));
    memset(layer, 0, sizeof(layer));
    memset(vv, 0, sizeof(vv));
    tot = 0;
}

void add(int u, int v, int w) {
    E[tot].v = v;
    E[tot].w = w;
    E[tot].nxt = head[u];
    head[u] = tot++;
}

int dis[MX], vis[MX];

void spfa(int s) {
    memset(dis, 0x3f, sizeof(dis));
    memset(vis, 0, sizeof(vis));
    queue<int>q;
    dis[s] = 0;//起点的值初始化为0；
    q.push(s);

    while(!q.empty()) {
        int u = q.front();
        q.pop();
        vis[u] = 0;

        for(int i = head[u]; ~i; i = E[i].nxt) {
            int v = E[i].v;
            if(dis[v] > dis[u] + E[i].w) {
                dis[v] = dis[u] + E[i].w;//不断以父节点来更新子节点的最短路；
                if(!vis[v]) {
                    vis[v] = 1;//如果没有入队，则将该点入队；
                    q.push(v);
                }
            }
        }
    }
}

int main() {
    //FIN;
    scanf("%d", &T);
    int cas = 1;
    while(T--) {
        init();
        scanf("%d%d%d", &n, &m, &c);
        for(int i = 1; i <= n; i++) {
            int x;
            scanf("%d", &x);
            layer[i] = x;//记录点i是位于第几层；
            vv[x] = 1;//记录第x层是否出现过点；
        }

        for(int i = 1; i < n; i++) {
            if(vv[i] && vv[i + 1]) {//只有相邻的两层均出现了点，才可以在这两层之间连边；
                add(n + i, n + i + 1, c);
                add(n + i + 1, n + i, c);
            }
        }

        for(int i = 1; i <= n; i++) {
            add(n + layer[i], i, 0);//点到自己的层的距离为0；
            if(layer[i] > 1)
                add(i, n + layer[i] - 1, c);
            if(layer[i] < n)
                add(i, n + layer[i] + 1, c);//将i点与相邻的层连接；
        }

        for(int i = 0; i < m; i++) {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            add(u, v, w);
            add(v, u, w);//再将普通的边连接起来；
        }
        printf("Case #%d: ", cas++);
        spfa(1);//spfa求解；
        if(dis[n] == INF)
            puts("-1");
        else
            printf("%d\n", dis[n]);
    }
    return 0;
}