题目链接:
题目大意:
n个点,m条边,很明确地说让求1 ~ n次短路。
数据范围:
1≤n≤50001≤m≤100000
1≤w≤5000(w表示每条边的边权)
解题思路:
次短路板子题。
Dijkstra算法就不再细说了。只需在Dijkstra算法内作一点修改,就可以求得次短路。
开两个数组,
dis[i]
表示源点到
i
的最短路,
-
如果D<dis[dr],那么在更新dr的最短路(即dis[dr])之前,先将当前的dis[dr]赋值给dist[dr],然后再更新dis[dr];
-
如果D>dis[dr]
&&
D<dist[dr]
,那么就直接将
D赋值给dist[dr]
。
值得注意的是,不管是更新了某个节点的最短路还是次短路后,都应该将这个节点的信息放进优先队列里,来继续更新其他的点。
详见代码
//Dijkstra次短路
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long LL;
const int inf = 1 << 30;
const LL INF = 1LL << 60;
const int MaxN = 1e5;
int n, m, all;
int pre[2 * MaxN + 5], last[MaxN + 5], other[2 * MaxN + 5];
int cost[2 * MaxN + 5];
int dis[MaxN + 5]; //最短路
int dist[MaxN + 5]; //次短路
struct Node
{
int id, d;
Node () {}
Node (int a, int b) {
id = a; d = b;
}
bool friend operator < (Node x, Node y) {
return x.d > y.d;
}
};
void build(int x, int y, int w)
{
pre[++all] = last[x];
last[x] = all;
other[all] = y;
cost[all] = w;
}
void Dijkstra(int s)
{
priority_queue <Node> pq;
for(int i = 1; i <= n; i++) {
dis[i] = inf;
dist[i] = inf;
}
dis[s] = 0;
Node tmp = Node(1, 0);
pq.push(tmp);
while(!pq.empty())
{
Node now = pq.top();
pq.pop();
int ed = last[now.id], dr;
if(dist[now.id] < now.d) continue;
while(ed != -1) {
dr = other[ed];
//把dis[now.id]改为now.d就过了!
//why!?
//哦!知道了!dis[]存的是最短路,后面更新次短路的时候会出错!
int D = now.d + cost[ed];
if(D < dis[dr]) { //更新dis[dr]和dist[dr]
dist[dr] = dis[dr];
dis[dr] = D;
tmp = Node(dr, dis[dr]);
pq.push(tmp);
}
else if(D > dis[dr] && D < dist[dr]) { //只更新dist[dr]
dist[dr] = D;
tmp = Node(dr, dist[dr]);
pq.push(tmp);
}
ed = pre[ed];
}
}
}
int main()
{
while(scanf("%d %d", &n, &m) != EOF)
{
all = -1;
memset(last, -1, sizeof(last));
for(int i = 1; i <= m; i++) {
int u, v, w;
scanf("%d %d %d", &u, &v, &w);
build(u, v, w); build(v, u, w);
}
Dijkstra(1);
printf("%d\n", dist[n]);
}
return 0;
}
这道也是次短路。