次短路 Roadblocks

题目链接
刚开始想了半天,想了一个(m+n)logn的做法,后来发现不能实现一条路可以走很多遍。代码如下:

#include <algorithm>
#include <bitset>
#include <cstdio>
#include <cstring>
#include <ctime>
#include <deque>
#include <map>
#include <iomanip>
#include <ios>
#include<iosfwd>
#include <iostream>
#include <istream>
#include <ostream>
#include <queue>
#include <set>
using namespace std;

typedef long long ll;
#define maxn 5010
ll n,r,cnt;ll head[maxn];
ll xx,yy,ww;ll d1[maxn];
ll d2[maxn];ll vis[maxn];
struct node{
	ll x,y,w,next;
}ed[200010];

void add(ll a,ll b,ll c){
	++cnt;
	ed[cnt].y=b;
	ed[cnt].w=c;
	ed[cnt].next=head[a];
	head[a]=cnt;
}


void dijkstra(){
	memset(d1,0x3f,sizeof(d1));
	memset(d2,0x3f,sizeof(d2));
	typedef pair<ll,ll>pii;
	priority_queue<pii,vector<pii>,greater<pii>>q;//优先弹出最小的 
	q.push(pii(0,1));d1[1]=0;//d2[1]=0;
	while(!q.empty()){
		ll uu=q.top().second;q.pop();//注意队列用的是top 
		if(vis[uu])continue;
		vis[uu]=1;
		for(int i=head[uu];i;i=ed[i].next){
			ll vv=ed[i].y;ll ww=ed[i].w;
			if(d1[uu]+ww<d1[vv]){
				d2[vv]=min(d1[vv],d2[uu]+ww);
				d1[vv]=d1[uu]+ww;
				q.push(pii(d1[vv],vv));
			}
			if(d1[uu]+ww>d1[vv]){
				d2[vv]=min(d2[vv],d1[uu]+ww);
//				printf("%lld %lld %lld\n",uu,vv,d1[uu]+ww);
			}
		}
    }
}

int main(){
	cin>>n>>r;
	for(int i=1;i<=r;++i){
		cin>>xx>>yy>>ww;
		add(xx,yy,ww);add(yy,xx,ww);
	}
	dijkstra();
	printf("%lld\n",d2[n]);
	return 0;
}

这就是基于dijksta算法的一个小改动,dijkstra算法是不会重复访问一个点的。但是比如loj的这组数据是行不通的:
5 10
1 2 3355
2 3 1712
1 4 4892
1 5 1442
5 3 4768
3 1 4334
2 1 4365
3 1 2591
2 1 3817
3 2 4068
这个算法需要一些小变动。提高算法的时间复杂度来获取正解。
正解如下:

#include <algorithm>
#include <bitset>
#include <cstdio>
#include <cstring>
#include <ctime>
#include <deque>
#include <map>
#include <iomanip>
#include <ios>
#include<iosfwd>
#include <iostream>
#include <istream>
#include <ostream>
#include <queue>
#include <set>
using namespace std;

typedef long long ll;
#define maxn 5010
ll n,r,cnt;ll head[maxn];
ll xx,yy,ww;ll d1[maxn];
ll d2[maxn];ll vis[maxn];
struct node{
	ll x,y,w,next;
}ed[200010];

void add(ll a,ll b,ll c){
	++cnt;
	ed[cnt].y=b;
	ed[cnt].w=c;
	ed[cnt].next=head[a];
	head[a]=cnt;
}

void dijkstra(){
	memset(d1,0x3f,sizeof(d1));
	memset(d2,0x3f,sizeof(d2));
	typedef pair<ll,ll>pii;
	priority_queue<pii,vector<pii>,greater<pii>>q;//优先弹出最小的 
	q.push(pii(0,1));d1[1]=0;//d2[1]=0;
	while(!q.empty()){
		ll uu=q.top().second;//优先队列用top 
		ll dd=q.top().first;q.pop();
		if(dd>d2[uu])continue;
		for(int i=head[uu];i;i=ed[i].next){
			ll vv=ed[i].y;ll ww=ed[i].w;
			if(d1[vv]>dd+ww){
				d2[vv]=d1[vv];
				d1[vv]=dd+ww;
				q.push(pii(d1[vv],vv));
			}
			if(d1[vv]<dd+ww&&d2[vv]>dd+ww){
				d2[vv]=dd+ww;
				q.push(pii(d2[vv],vv));
			}
		}
	}
}

int main(){
	cin>>n>>r;
	for(int i=1;i<=r;++i){
		cin>>xx>>yy>>ww;
		add(xx,yy,ww);add(yy,xx,ww);
	}
	dijkstra();
	printf("%lld\n",d2[n]);
	return 0;
} 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值