loj 1099(最短路）

最新推荐文章于 2021-10-10 23:11:47 发布

weixin_30568591

最新推荐文章于 2021-10-10 23:11:47 发布

阅读量65

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/wally/p/3361562.html

本文介绍了一种使用Dijkstra算法求解图中两点间次短路径的方法，并提供了完整的C++实现代码。该方法通过维护两个距离数组来记录每个节点的最短路径和次短路径。

题目链接：http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=25956

思路：dist[v][0]代表走到点v的最短路，dist[v][1]代表走到点v的次短路（dist[v][0]!=dist[v][1]),然后Dijkstra更新就可以了。

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 #include<queue>
 6 #include<vector>
 7 using namespace std;
 8 #define MAXN 5555
 9 #define MAXM 555555
10 #define inf 1<<30
11 #define FILL(a,b) memset(a,b,sizeof(a))
12 
13 struct Edge{
14     int v,w,next;
15 }edge[MAXM];
16 
17 int n,m,NE;
18 int head[MAXN];
19 
20 void Insert(int u,int v,int w)
21 {
22     edge[NE].v=v;
23     edge[NE].w=w;
24     edge[NE].next=head[u];
25     head[u]=NE++;
26 }
27 
28 int dist[MAXN][2];
29 int Dijkstra()
30 {
31     for(int i=1;i<=n;i++)
32         for(int j=0;j<=1;j++)dist[i][j]=inf;
33     priority_queue<pair<int,int>,vector<pair<int,int> >,greater<pair<int,int> > >que;
34     que.push(make_pair(0,1));
35     dist[1][0]=0;
36     while(!que.empty()){
37         pair<int,int>p=que.top();
38         que.pop();
39         for(int i=head[p.second];i!=-1;i=edge[i].next){
40             int v=edge[i].v,w=edge[i].w;
41             if(p.first+w<dist[v][0]){
42                 dist[v][1]=dist[v][0];
43                 dist[v][0]=p.first+w;
44                 que.push(make_pair(dist[v][0],v));
45             }else if(p.first+w!=dist[v][0]&&p.first+w<dist[v][1]){
46                 dist[v][1]=p.first+w;
47                 que.push(make_pair(dist[v][1],v));
48             }
49         }
50     }
51     return dist[n][1];
52 }
53 int main()
54 {
55     int _case,t=1;
56     scanf("%d",&_case);
57     while(_case--){
58         scanf("%d%d",&n,&m);
59         NE=0;
60         FILL(head,-1);
61         while(m--){
62             int u,v,w;
63             scanf("%d%d%d",&u,&v,&w);
64             Insert(u,v,w);
65             Insert(v,u,w);
66         }
67         printf("Case %d: %d\n",t++,Dijkstra());
68     }
69     return 0;
70 }