POJ 3255 Roadblocks(次短路模板题)

http://poj.org/problem?id=3255

写了还算多的最短路题目发现,没写过次短路。顺着挑战刷,就看到了次短路,然后发现一脸懵逼。

其实理解了后发现超级简单。次短路,每个点都记录两个距离就好了,最短的和次短的。然后最后输出第n个点的次短的即可。

代码如下:

#include<iostream>
#include<vector>
#include<utility>
#include<queue>
#include<cstdio>
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 5005;
typedef pair<int, int> pii;
int dis[maxn], dis_2[maxn];

struct node{
	int p, val;
	node(){}
	node(int a, int b){
		p = a;
		val = b;
	}
};

vector <node> edge[maxn];
inline void add_edge(int u, int v, int val){
	edge[u].push_back(node(v, val));
	edge[v].push_back(node(u, val));
}

void dijkstra(int s){
	fill(dis, dis + maxn, INF);
	fill(dis_2, dis_2 + maxn, INF);
	priority_queue<pii, vector<pii>, greater<pii> >q;
	q.push(make_pair(0, s));
	dis[s] = 0;
	while(!q.empty()){
		pii now = q.top();
		q.pop();
		if(now.first > dis_2[now.second])
			continue;
		for(int i = 0; i < edge[now.second].size(); i++){
			node next = edge[now.second][i];
			int d = now.first + next.val;
			if(dis[next.p] > d) {
				swap(dis[next.p], d);
				q.push(make_pair(dis[next.p], next.p));
			} 
			if(dis[next.p] < d && dis_2[next.p] > d) {
				dis_2[next.p] = d;
				q.push(make_pair(d, next.p));
			}
		}
	}
}

int main(){
	int n, r, u, v, val;
	cin >> n >> r;
	while(r--){
		scanf("%d%d%d", &u, &v, &val);
		add_edge(u, v, val);
	}
	dijkstra(1);
	
	cout << dis_2[n] << endl;
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值