题意:给定n个数,m条边,c是每一层的连接费用,n个数代表n层,每一个数字在一层中,从一层只能走到下一层或上一层的一个节点,费用为c,现在有m条边连接点,问你从1走到n的最小费用
思路:以两层为例,前边一堆的话代表第一层的每个点到第二层的所有点的费用都是c,其他边直接连即可,但是这么连肯定是超时的,所以我们可以每一层建立一个超级源点,这个源点到这层的每一个点费用都为0,而这些点可以连向相邻两层的超级源点费用为c,剩下的边正常连跑最短路即可
#include <queue>
#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=200010;
struct edge{
int to,cost;
edge(int a,int b){to=a;cost=b;}
};
typedef pair<int,int>P;
vector<edge>G[maxn];
int dis[maxn];
void dijkstra(int s){
priority_queue<P,vector<P>,greater<P> >que;
fill(dis,dis+maxn,inf);
dis[s]=0;que.push(P(0,s));
while(!que.empty()){
P p=que.top();que.pop();
int v=p.second;
if(dis[v]<p.first) continue;
for(unsigned int i=0;i<G[v].size();i++){
edge e=G[v][i];
if(dis[e.to]>dis[v]+e.cost){
dis[e.to]=dis[v]+e.cost;
que.push(P(dis[e.to],e.to));
}
}
}
}
int id[maxn];
bool vis[maxn];
int main(){
int T,n,m,k,u,v,c,cas=1,val;
scanf("%d",&T);
while(T--){
scanf("%d%d%d",&n,&m,&k);
for(int i=0;i<maxn;i++) G[i].clear(),vis[i]=0;
for(int i=1;i<=n;i++){
scanf("%d",&val);
id[i]=val;vis[val]=1;
}
for(int i=1;i<=n;i++){
if(id[i]==1){
G[id[i]].push_back(edge(i+n,0));
if(vis[id[i]+1]==1&&id[i]<n) G[i+n].push_back(edge(id[i]+1,k));
}else if(id[i]==n){
G[id[i]].push_back(edge(i+n,0));
if(vis[id[i]-1]==1&&id[i]>1) G[i+n].push_back(edge(id[i]-1,k));
}else{
G[id[i]].push_back(edge(i+n,0));
if(vis[id[i]+1]==1) G[i+n].push_back(edge(id[i]+1,k));
if(vis[id[i]-1]==1) G[i+n].push_back(edge(id[i]-1,k));
}
}
for(int i=0;i<m;i++){
scanf("%d%d%d",&u,&v,&c);
G[u+n].push_back(edge(v+n,c));
G[v+n].push_back(edge(u+n,c));
}
dijkstra(n+1);
if(dis[2*n]==inf) printf("Case #%d: -1\n",cas++);
else printf("Case #%d: %d\n",cas++,dis[2*n]);
}
return 0;
}