迪杰斯特拉算法应用1——输出所有顶点的最短距离

输入样例:
第一行:顶点数、边数、起始顶点编号
第2-边数+1行:顶点a,顶点b,边权

输出:输出所有顶点的最短距离
注:为有向带权图
代码:

#include <stdio.h>
#include <algorithm>
using namespace std;

const int maxn = 1000;
const int INF = 100000000;

int n,m,s,G[maxn][maxn];
int d[maxn];
bool vis[maxn] = {false};

void Dijkstra(int s){
	fill(d,d+maxn,INF);
	d[s] = 0;
	for(int i = 0;i <n;i++){
		int u = -1,MIN = INF;
		for(int j = 0;j <n;j++){
			if(vis[j] == false && d[j]<MIN){
				u = j;
				MIN = d[j];
			} 
		}
		if(u == -1) return;
		vis[u] = true;
		for(int v = 0; v<n;v++){
			if(vis[v] == false && G[u][v] != INF && d[u]+G[u][v] <d[v]){
				d[v] = d[u]+G[u][v];
			}
		}
	}
}

int main(){
	int u,v,w;
	scanf("%d %d %d",&n,&m,&s);    //顶点个数,边数,起点编号 
	fill(G[0],G[0]+maxn*maxn,INF);    //初始化图G 
	for(int i = 0;i <m;i++){
		scanf("%d%d%d",&u,&v,&w);    //输入u,v,以及u->v的边权 
		G[u][v] = w;
	}
	Dijkstra(s);        //算法入口 
	for(int i = 0;i <n;i++){
		printf("%d ",d[i]);   //输出所有顶点的最短距离 
	}
	return 0; 
}

/*
6 8 0
0 1 1
0 3 4
0 4 4
1 3 2
2 5 1
3 2 2
3 4 3
4 5 3
*/
发布了212 篇原创文章 · 获赞 6 · 访问量 6418

猜你喜欢

转载自blog.csdn.net/weixin_42377217/article/details/104237559
今日推荐