题目链接
刚开始想了半天,想了一个(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;
}