题目速递:http://acm.hdu.edu.cn/showproblem.php?pid=4725
题意:多case;
有n个点,每个点属于一个层。相邻层次的点互相可达,边权为c,除此之外有另外的m条边,边权可指定。求点1到点n的最短距离。
题解:按照常规做法,直接各自连边再跑一次最短路即可。但是两个相邻层次间的连边的次数会是该相邻层次点数乘积,不可行。所以建图是关键。
给每个层次建一个入点和出点,层次间的连边就只需要将前后层次的出点和入点连接起来就行了。每个点编号[1,n],各层次入点编号在[1+n,n+n]之间,出点在[1+2n,3n]之间。其他部分常规操作。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<cstring>
#define ll long long
using namespace std;
const int maxn=3e5+5;
struct node{
int id,w;
};
vector<node>G[maxn];
const int inf=0x3f3f3f3f;
struct qnode{
int v;
int c;
bool operator <(const qnode &r)const{
return c>r.c;
}
};
int node_num;
int vis[maxn];
int dist[maxn];
void init(){
for(int i=1;i<=maxn;i++)G[i].clear();
}
void Dijkstra(int start)
{
memset(vis,0,sizeof(vis));
memset(dist,inf,sizeof(dist));
priority_queue<qnode>qu;
while(!qu.empty())qu.pop();
dist[start]=0;
qu.push((qnode){start,0});
qnode tmp;
while(!qu.empty())
{
tmp=qu.top();qu.pop();
int u=tmp.v;
if(vis[u])continue;
vis[u]=1;
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i].id;
int cost=G[u][i].w;
if(dist[v]>dist[u]+cost)
{
dist[v]=dist[u]+cost;
if(!vis[v])
qu.push((qnode){v,dist[v]});
}
}
}
}
int main(){
int T;
cin>>T;
int t=0;
int n,m,c;
while(T--){
t++;
scanf("%d%d%d",&n,&m,&c);
int layer;
node_num=0;
for(int i=1;i<=n;i++){
scanf("%d",&layer);
node_num=max(node_num,layer);
G[layer+n].push_back((node){i,0});
G[i].push_back((node){layer+2*n,0});
}
for(int i=1;i<node_num;i++){
G[i+2*n].push_back((node){(i+1)+n,c});
G[(i+1)+2*n].push_back((node){i+n,c});
}
for(int i=1;i<=m;i++){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
G[u].push_back((node){v,w});
G[v].push_back((node){u,w});
}
Dijkstra(1);
if(dist[n]==inf)dist[n]=-1;
printf("Case #%d: %d\n",t,dist[n]);
init();
}
}