HDU 4725 The Shortest Path in Nya Graph

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

题意：

 

一共有N个点，每个点都在一个楼层里，相邻楼层之间可以走，权值为C（注意：同一层楼之间的权值不是0）。另外有m条无向边连接两个点。问从点1到点n的最短路径。

 

题解：

题目的范围有点大，不能直接建边。可以将每一层抽象出一个点，也就是说一共有2*n个点。如果点A在楼层B上，那么就建一条楼层B到点A的单向边，权值为0。对于所有的点都与和它相邻的楼层建一条权值为C的双向边。多出来的m条边，就正常的建双向边就好了。

#include <iostream>
#include <cstdio>
#include <string>
#include <string.h>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
using namespace std;
#define LL long long
struct Edge{
    int to,val,next;
    Edge(int to=0,int val=0,int next=0):to(to),val(val),next(next){}
}e[500005];
int head[200005],now;
int T,n,m,c;
void add(int u,int v,int w){
    e[++now]=Edge(v,w,head[u]);head[u]=now;
}
int dis[200005],vis[200005];
struct Node{
    int id,val;
    Node(int id,int val):id(id),val(val){}
    bool operator<(const Node a)const{
        return this->val>a.val;
    }
};
int a[200005];
void dij(){
    memset(dis,125,sizeof(dis));
    memset(vis,0,sizeof(vis));
    priority_queue<Node>q;
    q.push(Node(1,0));
    dis[1]=0;
    while(!q.empty()){
        Node t=q.top();q.pop();
        int u=t.id;
        vis[u]=1;
        for(int i=head[u];~i;i=e[i].next){
            int v=e[i].to;
            if(!vis[v]&&dis[v]-dis[u]>e[i].val){
                dis[v]=dis[u]+e[i].val;
                q.push(Node(v,dis[v]));
            }
        }
    }
}
int main(){
    scanf("%d",&T);
    for(int t=1;t<=T;t++){
        scanf("%d%d%d",&n,&m,&c);
        memset(head,-1,sizeof(head));now=-1;
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            add(n+a[i],i,0);
            if(a[i]>1) add(i,n+a[i]-1,c);
            if(a[i]<n) add(i,n+a[i]+1,c);
        }
        int u,v,w;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
            add(v,u,w);
        }
        dij();
        if(dis[n]>(int)1e9) dis[n]=-1;
        printf("Case #%d: %d\n",t,dis[n]);
    }
}