301 最短路 Dijkstra 演算法
阿新 • • 發佈:2022-05-28
視訊連結:
//////////////////////Dijkstra #include <iostream> #include <cstring> #include <algorithm> #include <vector> #define inf 2147483647 using namespace std; int n,m,s,a,b,c; const int N=100010; struct edge{int v,w;}; vector<edge> e[N]; int d[N], vis[N]; void dijkstra(int s){ for(int i=0;i<=n;i++)d[i]=inf; d[s]=0; for(int i=1;i<n;i++){//列舉次數 int u=0; for(int j=1;j<=n;j++)//列舉點 if(!vis[j]&&d[j]<d[u]) u=j; vis[u]=1; //標記u已出圈 for(auto ed:e[u]){//列舉鄰邊 int v=ed.v, w=ed.w; if(d[v]>d[u]+w){ d[v]=d[u]+w; } } } } int main(){ cin>>n>>m>>s;for(int i=0; i<m; i++){ cin>>a>>b>>c; e[a].push_back({b,c}); } dijkstra(s); for(int i=1;i<=n;i++) printf("%d ",d[i]); return 0; } ///////////////////Heap-Dijkstra #include <cstring> #include <iostream> #include <algorithm> #include <queue> #define inf 2147483647 usingnamespace std; const int N=100010; int n,m,s,a,b,c; struct edge{int v,w;}; vector<edge> e[N]; int d[N], vis[N]; priority_queue<pair<int,int>> q; void dijkstra(int s){ for(int i=0;i<=n;i++) d[i]=inf; d[s]=0; q.push({0,s}); while(q.size()){ auto t=q.top(); q.pop(); int u=t.second; if(vis[u])continue;//再出隊跳過 vis[u]=1;//標記u已出隊 for(auto ed : e[u]){ int v=ed.v, w=ed.w; if(d[v]>d[u]+w){ d[v]=d[u]+w; q.push({-d[v],v});//大根堆 } } } } int main(){ cin>>n>>m>>s; for(int i=0; i<m; i++){ cin>>a>>b>>c; e[a].push_back({b,c}); } dijkstra(s); for(int i=1;i<=n;i++) printf("%d ",d[i]); return 0; }