1. 程式人生 > 其它 >301 最短路 Dijkstra 演算法

301 最短路 Dijkstra 演算法

視訊連結:

//////////////////////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 using
namespace 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; }