堆优化Dijkstra计算最短路+路径计数
今天考试的时候遇到了一道题需要路径计数,然而蒟蒻从来没有做过,所以在考场上真的一脸懵逼。然后出题人NaVi_Awson说明天考试还会卡SPFA,吓得我赶紧又来学一波堆优化的Dijkstra(之前只会SPFA。。。
堆优化Dijkstra
其实Dijkstra的思想很简单。SPFA是以边为基础的最短路松弛,那么Dijkstra恰好相反,是以点为基础的最短路松弛。划分两个点的集合,一个是已经松弛的点集合,一个是未松弛的点集合,每次从已松弛的点集合中找当前路径最小的点来松弛与它相连的未松弛的点。但是如果是不加优化的Dijkstra复杂度是$O(n^2)$的,肯定会T,所以这里引入堆优化,复杂度可以降到$O((n+m)\log n)$。
路径计数
路径计数其实也很简单,在做最短路的时候多引入一个数组,表示当前从远点到达该点的最短路径条数。松弛的时候如果发现该点当前的最短路长度和它被松弛得到的长度相等,那么说明到达它的最短路径条数又增加了,那么就把该点的路径条数加上松弛点的路径条数;如果该点的最短路长度小于被松弛的长度,那么就更新最短路,同时更新路径条数,令该点的最短路径条数等于松弛点的路径条数。
最后再放上自己打的堆优化Dijkstra最短路+路径计数代码。
Code:
//It is made by HolseLee on 8th Aug 2018 //Dijkstra #include<bits/stdc++.h> using namespace std; typedef long long ll; const int N=1e5+7; int n,m,sta,size,hd[N]; ll dis[N]; bool vis[N]; struct Edge { int to,nxt; ll val; }edge[N<<2]; struct Node { int x; ll val; bool operator < (const Node a) const { return val>a.val; } }; priority_queue<Node>t; inline void add(int x,int y,ll v) { edge[++size].to=y; edge[size].val=v; edge[size].nxt=hd[x]; hd[x]=size; } void dijkstra() { memset(vis,false,sizeof(vis)); memset(dis,0x7f,sizeof(dis)); dis[sta]=0; t.push((Node){sta,dis[sta]}); int x,y; while( !t.empty() ) { x=t.top().x; t.pop(); if( vis[x] ) continue; vis[x]=true; for(int i=hd[x]; i!=-1; i=edge[i].nxt) { y=edge[i].to; if( dis[y]>dis[x]+edge[i].val ) { dis[y]=dis[x]+edge[i].val; t.push((Node){y,dis[y]}); } } } } int main() { std::ios::sync_with_stdio(false); memset(hd,-1,sizeof(hd)); cin>>n>>m>>sta; int x,y; ll v; for(int i=1; i<=m; i++) { cin>>x>>y>>v; if( x==y ) continue; add(x,y,v); } dijkstra(); for(int i=1; i<=n; i++) cout<<dis[i]<<" "; return 0; }
蒟蒻写博客不易,如果有误还请大佬们提出
如需转载,请署名作者并附上原文链接,蒟蒻非常感激
名称:HolseLee
博客地址:www.cnblogs.com/cytus
个人邮箱:1073133650@qq.com
如需转载,请署名作者并附上原文链接,蒟蒻非常感激
名称:HolseLee
博客地址:www.cnblogs.com/cytus
个人邮箱:1073133650@qq.com