dijkstra模板(易理解)

dijkstra模板(优先队列+链式前向星)

#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f; 
const int maxn=2e3+10;
int head[maxn],cnt;
struct Edge{
	int to,next,w;
}edge[maxn];
void add_Edge(int u,int v,int w){
	edge[cnt].to=v;
	edge[cnt].w=w;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
void init(){
	memset(head,-1,sizeof head);
	cnt=0;
}
struct s_node{
	int id,n_dis;				//当前节点的id和起点到该点的距离 
	s_node(int b,int c){id=b;n_dis=c;}
	friend bool operator<(const s_node &a,const s_node &b)
	{return a.n_dis>b.n_dis;}
};
int n,m,pre[maxn];
void print_path(int s,int t){	//打印s到t的最短路径 
	if(s==t){
		printf("%d",s);
		return;
	}
	print_path(s,pre[t]);
	printf("%d",t);
}
void dijkstra(){
	int s=1;					//设置起点 
	int dis[maxn];				//记录该点到起点的距离 
	bool done[maxn];			//记录该点是否已经找到最短距离 
	for(int i=1;i<=n;++i) {dis[i]=INF;done[i]=false;}	//初始化
	dis[s]=0;					//起点到自己的距离为0 
	priority_queue<s_node>que;	//优先队列存储节点 
	que.push(s_node(s,dis[s]));
	while(!que.empty()){
		s_node u=que.top();
		que.pop();
		if(done[u.id]) continue;//该点已经是最短路径,不需要寻找 
		done[u.id]=true;
		for(int i=head[u.id];~i;i=edge[i].next){		//遍历该点邻居 
			Edge y=edge[i];
			if(done[y.to]) continue;
			if(dis[y.to]>y.w+u.n_dis){					//从当前节点绕路到邻居更近 
				dis[y.to]=y.w+u.n_dis;
				que.push(s_node(y.to,dis[y.to]));
				pre[y.to]=u.id;	//可记录路径 
			}
		}
	}
	print_path(s,n);			//输出路径 
}
int main(void){
	while(~scanf("%d %d",&n,&m)){
		if(n==0&&m==0) return 0;
		init();
		while(m--){							//链式前向星建图 
			int a,b,c;
			scanf("%d %d %d",&a,&b,&c);
			add_Edge(a,b,c);
			add_Edge(b,a,c);
		}
		dijkstra();
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值