题意:
给若干点,分布于若干层,同层的点是不可以直接到达的,邻层的点可以相互到达,花费为c,同时给了若干无向路,求1到n的最小花费
分析:
题目难点在于建图,要将层抽象为边的关系,于是我们可以假设出2 * n个点,每层一个进点,一个出点,进点有到该层的实点和相邻层的出点到它的有向路,出点反之
本题去了一个坑点,题意上相邻层若没点是不可互通的,也就是说我们需要考虑一个无点层两侧的点的关系,但是题目所给的输入时排除了这种情况的
另外因为是稠密图,所以迪杰斯特拉+堆优化的效率更高,spfa+堆优化勉强能过。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6;
const int inf=0x3f3f3f3f;
int head[maxn];
int tot;
struct Edge{
int to,w;
int nxt;
Edge(){}
Edge(int v,int w):to(v),w(w){}
}edge[maxn];
void addedge(int u,int v,int w){
edge[tot].to=v;
edge[tot].w=w;
edge[tot].nxt=head[u];
head[u]=tot++;
}
struct qnode{
int v,c;
qnode(int v=0,int c=0):v(v),c(c){}
bool operator < (const qnode &r) const {
return c>r.c;
}
};
bool vis[maxn];
int dis[maxn];
void dijkstra(int s,int n){
memset(vis,0,sizeof vis);
for(int i=1;i<=n;i++){
dis[i]=inf;
}
priority_queue<qnode> q;
dis[s]=0;
q.push(qnode(s,0));
qnode tmp;
while(!q.empty()){
tmp=q.top();
q.pop();
int u=tmp.v;
if(vis[u]) continue;
vis[u]=true;
for(int i=head[u];i!=-1;i=edge[i].nxt){
int v=edge[i].to;
int w=edge[i].w;
if(!vis[v]&&dis[v]>dis[u]+w){
dis[v]=dis[u]+w;
q.push(qnode(v,dis[v]));
}
}
}
}
int main(){
int T;
scanf("%d",&T);
for(int cs=1;cs<=T;cs++){
int n,m,c;
scanf("%d%d%d",&n,&m,&c);
memset(head,-1,sizeof head);
tot=0;
for(int i=1;i<=n;i++){
int r;
scanf("%d",&r);
addedge(n+r,i,0);
addedge(i,2*n+r,0);
}
while(m--){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
addedge(u,v,w);
addedge(v,u,w);
}
for(int i=1;i<n;i++){
addedge(n*2+i,n+i+1,c);
addedge(n*2+i+1,n+i,c);
}
dijkstra(1,3*n);
if(dis[n]==inf) dis[n]=-1;
printf("Case #%d: %d\n",cs,dis[n]);
}
return 0;
}