Dijkstra 算法最短路模板

Dijkstra 算法

从单个源点出发，到所有节点的最短路
#include<iostream>
#include<vector>
#include<cstring>
#include<queue>
#define maxn 100
#define INF 100000
using namespace std;
struct Edge{
	int from,to,dis;
	Edge(int u,int v,int d):from(u),to(v),dis(d){}
};
struct HeapNode{//最小d值和对应的结点编号 
	int d,u;
	bool operator <(const HeapNode &rhs) const{
		return d>rhs.d;
	}  
};
struct Dijkstra{
	int n,m;
	vector<Edge> edge;//存放图中各条边的情况 
	vector<int> G[maxn];//从G[u]出发对应的边的编号，方便寻找到各边
	bool done[maxn];//是否已访问 
	int d[maxn];//源点到各边的情况 
	int p[maxn];//最短路中的一条边
	void init(int n){//初始化 
		this->n=n;
		for(int i=0;i<n;i++)  G[i].clear();
		edge.clear();
	}
	void AddEdge(int from,int to,int dis){//添加边 
		edge.push_back(Edge(from,to,dis));//如果是无向图，需要调用两次
		m=edge.size();
		G[from].push_back(m-1);
	}
	void dijkstra(int s){//求s到所有点的距离
		priority_queue<HeapNode> Q;
		for(int i=0;i<n;i++) d[i]=INF;
		d[s]=0;
		memset(done,0,sizeof(done));
		Q.push(HeapNode{0,s});
		while(!Q.empty()){
			HeapNode x=Q.top();Q.pop();
			int u=x.u;
			if(done[u]) continue;
			done[u]=true; 
			for(int i=0;i<G[u].size();i++){
				Edge &e=edge[G[u][i]]; 
				if(d[e.to]>d[u]+e.dis){
				    d[e.to]=d[u]+e.dis;
				    p[e.to]=G[u][i];
				    Q.push(HeapNode{d[e.to],e.to});
				}
			}
		}
	}
	
};
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	Dijkstra dt;
	dt.init(n);
	for(int i=0;i<m;i++){
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		dt.AddEdge(a,b,c);
	}
	dt.dijkstra(0);
	for(int i=0;i<n;i++){
		cout<<dt.d[i]<<endl;
	}
	
	return 0;
}