9.9单源最短路径问题、Dijkstra算法+模板代码

---

一、单源最短路径问题

单源最短路径问题（Single Source Shortest Path,SSSP问题）：设 1 号点为起点，求长度为 n（ [ 1, n]为节点之间的连续编号）的数组 dist, 其中，dist [i] 表示从起点1到 节点 i 的最短路径的长度

二、Dijkstra算法

算法流程

1.初始化 dist [1] =0,其余节点的 dist 值为正无穷大；
2.找出一个未被标记的、dist [x] 最小的节点 x，然后标记节点 x；
3.扫描节点 x的所有出边（x, y，z），若 dist [x] +z < dist [y]，则使用 dist [x] +z 更新 dist [y]
4.重复 第2、3步，直至所有节点都被打上标记 。

Dijkstra算法是基于贪心思想的，只适用于所有边的长度都是非负数的图

//时间复杂度为O(n^2) 
void dijkstra(){
	memset(d,0x3f,sizeof(d));//dist 数组 
	memset(v,0,sizeof(v));//节点标记 
	d[1]=0;
	for(int i=1;i<n;i++){//重复n-1次 
		int x=0;
		//找到未标记节点中dist最小的 
		for(int j=1;j<=n;j++){
			if(!v[i]&&(x==0||d[j]<d[x])) x=j;
		}
		v[x]=1;
		//用全局最小值点 x 更新其他节点 
		for(int y=1;y<=n;y++){
			d[y]=min(d[y],d[x]+a[x][y]);
		}
	}
}
int main(){
	cin>>n>>m;
	//构建邻接矩阵 
	memset(a,0x3f,sizeof(a));
	for(int i=1;i<=n;i++) a[i][i]=0;//自己到自己的矩阵点为0 
	for(int i=1;i<=m;i++){
		int x,y,z;
		cin>>x>>y>>z;
		a[x][y]=min(a[x][y],z);
	} 
	dijkstra();
	for(int i=1;i<=n;i++) cout<<d[i]<<endl;
} 

优化：利用二叉堆对dist数组进行维护，用 O(logn)的时间获取最小值并从堆中删除，用O(logn)的时间执行一条边的扩展和更新，最终在O ( m*logn )的时间实现Dijkstra算法

//时间复杂度为O(m*logn) 
priority_queue<pair<int,int> >q;//大根堆
//pair的第一维为dist的相反数（利用相反数变成小根堆），第二维为节点编号 
void add(int x,int y,int z){
	ver[++tot]=y;edge[tot]=z;
	next[tot]=head[x];
	head[x]=tot;
}
void dijkstra(){
	memset(d,0x3f,sizeof(d));//dist 数组 
	memset(v,0,sizeof(v));//节点标记 
	d[1]=0;
	q.push(make_pair(0,1)); 
	while(q.size()){
		//取出堆顶 
		int x=q.top().second;q.pop();
		if(v[x]) continue;
		v[x]=1;
		//扫描所有出边 
		for(int i=head[x];i;i=next[i]){
			int y=ver[i],z=edge[i];
			if(d[y]>d[x]+z){
				//更新，把新的二元组插入堆 
				d[y]=d[x]+z;
				q.push(make_pair(-d[y],y));
			}
		} 
	}
}
int main(){
	cin>>n>>m;
	//构建邻接表 
	memset(a,0x3f,sizeof(a));
	for(int i=1;i<=m;i++){
		int x,y,z;
		cin>>x>>y>>z;
		add(x,y,z);
	} 
	dijkstra();
	for(int i=1;i<=n;i++) cout<<d[i]<<endl;
}