1. 程式人生 > 實用技巧 >洛谷 P4779 【模板】單源最短路徑(標準版)

洛谷 P4779 【模板】單源最短路徑(標準版)

單源最短路徑,非負權圖,所以可以用Dijkstra;

具體看程式碼:

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<vector>
 4 #include<queue>
 5 #include<algorithm>
 6 #include<cstring>
 7 #include<cmath>
 8 using namespace std;
 9 const int maxn=500010;
10 struct zkw{
11     int to;
12 int cost; 13 }; 14 vector<zkw> k[maxn]; //vector模擬鄰接表 15 priority_queue< pair<int,int> > q; //模擬堆,用堆優化的Dijkstra 16 int dis[maxn],vis[maxn]; 17 void add(int x,int y,int z){ //加邊函式 18 k[x].push_back((zkw){y,z}); 19 } 20 void dij(){ //最短路核心函式 21 while(q.size()){ 22 int
x=q.top().second; //不斷尋找權值最小的邊 23 q.pop(); 24 if(vis[x]) continue; 25 vis[x]=true; 26 for(int i=0;i<k[x].size();++i){ // 找到後對其他點進行更新 27 int y=k[x][i].to,z=k[x][i].cost; 28 if(dis[y]>dis[x]+z){ 29 dis[y]=dis[x]+z; 30 q.push(make_pair(-1
*dis[y],y)); 31 } 32 } 33 } 34 } 35 int n,m,fm,a,b,c; 36 int main(){ 37 memset(dis,0x3f3f3f3f,sizeof(dis)); 38 memset(vis,0,sizeof(vis)); 39 scanf("%d%d%d",&n,&m,&fm); 40 dis[fm]=0; 41 for(int i=1;i<=m;++i){ 42 scanf("%d%d%d",&a,&b,&c); 43 add(a,b,c); 44 } 45 q.push(make_pair(0,fm)); 46 dij(); 47 for(int i=1;i<=n;++i){ 48 if(dis[i]==0x3f3f3f3f){ 49 printf("2147483647 "); 50 } 51 else printf("%d ",dis[i]); 52 } 53 return 0; 54 }

半倍經驗:

P3371 【模板】單源最短路徑(弱化版)