题目链接:
题意:
给定n个节点,m条带权单向路,特殊节点x,求所有节点到达x再返回的距离最大值。
解题思路:
n范围1k,多源的floyd会超时,考虑dij或者spfa。首先从x返回各个节点的距离就是dij跑出来的dis数组,问题在于计算各个节点前往x的距离,因为是单向路,这个距离和dij跑出的dis是不一样的。所以考虑反向建图,正向图反向图分别跑一遍dij算法,将两个dis数组求和,枚举找到最大值输出即可。
代码:
#include<iostream>
#include<cstring>
#include<queue>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl "\n"
using namespace std;
typedef long long ll;
typedef pair <int,int> P;
const int maxn=1005;
const int maxe=100000;
const int inf=0x3f3f3f3f;
int head1[maxn],head2[maxn];
struct Edge{
int to;
int next;
int w;
} edge1[maxe],edge2[maxe];//两张图
int cnt;
int dis2[maxn];
int dis[maxn];
void init(){
cnt=0;
memset(head1,-1,sizeof(head1));
memset(head2,-1,sizeof(head2));
return ;
}
inline void add(int u,int v,int w){//正向图正向边,反向图反向边
edge1[cnt].next=head1[u];
edge1[cnt].to=v;
edge1[cnt].w=w;
head1[u]=cnt;
edge2[cnt].next=head2[v];
edge2[cnt].to=u;
edge2[cnt].w=w;
head2[v]=cnt;
cnt++;
}
void dij1(int start){//正向图dij
memset(dis,0x3f,sizeof(dis));
priority_queue<P,vector<P>,greater<P> > q;
dis[start]=0;
q.push(P(0,start));
while(!q.empty()){
P p=q.top(); q.pop();
int v=p.second;
if(dis[v]<p.first) continue;
for(int i=head1[v];i!=-1;i=edge1[i].next){
int tmp=edge1[i].to;
if(dis[tmp]>dis[v]+edge1[i].w){
dis[tmp]=dis[v]+edge1[i].w;
q.push(P(dis[tmp],tmp));
}
}
}
return ;
}
void dij2(int start){//反向图dij
memset(dis2,0x3f,sizeof(dis2));
priority_queue<P,vector<P>,greater<P> > q;
dis2[start]=0;
q.push(P(0,start));
while(!q.empty()){
P p=q.top(); q.pop();
int v=p.second;
if(dis2[v]<p.first) continue;
for(int i=head2[v];i!=-1;i=edge2[i].next){
int tmp=edge2[i].to;
if(dis2[tmp]>dis2[v]+edge2[i].w){
dis2[tmp]=dis2[v]+edge2[i].w;
q.push(P(dis2[tmp],tmp));
}
}
}
return ;
}
int n,m,x;
int main(){
IOS
init();
cin>>n>>m>>x;
while(m--){
int u,v,w;
cin>>u>>v>>w;
add(u,v,w);
}
dij1(x),dij2(x);
int ans=-1;
for(int i=1;i<=n;i++)
if(dis[i]!=inf&&dis2[i]!=inf&&dis[i]+dis2[i]>ans)
ans=dis[i]+dis2[i];
cout<<ans<<endl;
return 0;
}