HDU4725 The Shortest Path in Nya Graph dij

分析：对于每一层，原来n个点，然后扩展为原来的三倍，每一层扩展一个入点，一个出点，然后跑最短路

注：tmd我把一个n写成m了，然后wa了7次，我都要怀疑人生了

#include<cstdio>
#include<cstring>
#include<queue>
#include<cstdlib>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
typedef long long LL;
const int N=1e5+;
const int INF=0x3f3f3f3f;
struct Edge{
   int v,w,next;
   bool operator<(const Edge &e)const{
      return w>e.w;
   }
}edge[N*];
int head[N*],tot,n,m,c,d[N*];
void add(int u,int v,int w){
   edge[tot].v=v;
   edge[tot].w=w;
   edge[tot].next=head[u];
   head[u]=tot++;
}
priority_queue<Edge>q;
bool vis[N*];
int dij(int s,int t){
    for(int i=;i<=n;++i)d[i]=INF,vis[i]=;
    d[s]=,q.push(Edge{s,,});
    while(!q.empty()){
       while(!q.empty()&&vis[q.top().v])q.pop();
       if(q.empty())break;
       int u=q.top().v;
       q.pop();
       vis[u]=;
       for(int i=head[u];~i;i=edge[i].next){
          int v=edge[i].v;
          if(!vis[v]&&d[v]>d[u]+edge[i].w){
            d[v]=d[u]+edge[i].w;
            q.push(Edge{v,d[v],});
          }
       }
    }
    return d[t]==INF?-:d[t];
}
vector<int>g[N];
int main(){
    int T,cas=;
    scanf("%d",&T);
    while(T--){
       scanf("%d%d%d",&n,&m,&c);
       for(int i=;i<=n;++i)g[i].clear();
       for(int i=;i<=n;++i){
        int x;
        scanf("%d",&x);
        g[x].push_back(i);
       }
       memset(head,-,sizeof(head));
       tot=;
       for(int i=;i<=m;++i){
         int u,v,w;
         scanf("%d%d%d",&u,&v,&w);
         add(u,v,w),add(v,u,w);
       }
       for(int i=;i<=n;++i){
        if(!g[i].size())continue;
        if(i>&&g[i-].size())
          add(n+i,*n+i-,c);
        if(i<n&&g[i+].size())
          add(n+i,*n+i+,c);
         for(int j=;j<g[i].size();++j)
            add(g[i][j],i+n,),add(i+*n,g[i][j],);
       }
       n*=;
       printf("Case #%d: %d\n",++cas,dij(,n/));
    }
    return ;
}