dijkstra算法求次短路径

今天打牛客网比赛的时候回顾了一道题,求次短路径,总结一下代码,以后复习方便用,这题思想还不错:
直接上代码:

#include <iostream>
#include <string.h>
#include <queue>
#include <algorithm>
#include <vector>
using namespace std;
#define N 5005
#define INF 111111111               //无限大表示两边之间没有路径
struct Edge{
	int to ,w;
	bool operator <(const Edge &a)const{
		return w > a.w;
	}
};
priority_queue<Edge> Q;
vector<Edge> V[N];
int n, m;
int dis[N], dis2[N];
void init(){
	for(int i = 1; i <= n; i++){
		V[i].clear();
		dis[i] = INF;
		dis2[i] = INF;
	}
}
int dijkstra(){          //使用dijkstra算法
	dis[1] = 0;
	Edge p, q, r;
	p.to = 1, p.w = 0;
	Q.push(p);
	while(!Q.empty()){
		p = Q.top();
		Q.pop();
		int u = p.to;
		if(p.w > dis2[u])continue;
		for(int i = 0; i < V[u].size(); i++){
			q = V[u][i];
			int to = q.to, d = q.w + p.w;
			if(dis[to] > d){
				swap(dis[to], d);
				r.to = to, r.w = dis[to];
				Q.push(r);
			}
			if(dis[to] < d && dis2[to] > d){
				dis2[to] = d;
				r.to = to, r.w = d;
				Q.push(r);
			}
		}
	}
	return dis2[n];
}
int main(){
	int a, b, w;
	Edge p;
	while(cin>>n>>m){
		init();
		for(int i = 0; i < m; i++){
			scanf("%d%d%d", &a, &b, &w);
			p.to = b;
			p.w = w;
			V[a].push_back(p);
			p.to = a;
			V[b].push_back(p);
		}
		int len = dijkstra();
		cout<<len<<endl;
	}
	return 0;
}

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值