【Codeforces666B】【World Tour】【最短路】

题目大意

给出一幅有向图，求出四个不同的点，两点之间走最短路，用最长的路走完四个点。

解题思路

跑n遍单源最短路，保存每个点最远的三个前驱，最远的三个后继。枚举中间两个点，再枚举它们的前驱和后继，判一下重，取最优答案。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=3000,maxm=5000;
int n,m,ans[10],gra[10],to[10][maxm+10],next[10][maxm+10],begin[10][maxn+10],dis[3][maxn+10][maxn+10],
f[10][maxn+10][10],queue[maxn+10];
bool inqueue[maxn+10];
void insert(int t,int u,int v){
to[t][++gra[t]]=v;
next[t][gra[t]]=begin[t][u];
begin[t][u]=gra[t];
}
struct rec{
int pon,dis;
friend bool operator<(rec x,rec y){
return (x.dis>y.dis)||((x.dis==y.dis)&&(x.pon<y.pon));
}
};
rec a[maxn+10];
void spfa(int t,int now){
int head=0,tail=0,inf=dis[0][0][0];
dis[t][now][now]=0;
inqueue[queue[++tail]=now]=1;
for(;head!=tail;){
int nown=queue[++head];
inqueue[nown]=0;
for(int i=begin[t][nown];i;i=next[t][i])
if(dis[t][now][to[t][i]]>dis[t][now][nown]+1){
dis[t][now][to[t][i]]=dis[t][now][nown]+1;
if(!inqueue[to[t][i]])inqueue[queue[++tail]=to[t][i]]=1;
}
}
fo(i,1,n){
a[i].pon=i;
a[i].dis=(dis[t][now][i]!=inf)?dis[t][now][i]:-inf;
}
sort(a+1,a+n+1);
fo(i,1,3)f[t][now][i]=a[i].pon;
}
int main(){
freopen("d.in","r",stdin);
freopen("d.out","w",stdout);
scanf("%d%d",&n,&m);
fo(i,1,m){
int u,v;scanf("%d%d",&u,&v);
insert(1,u,v);insert(2,v,u);
}
memset(dis,63,sizeof(dis));
fo(i,1,n){
spfa(1,i);
spfa(2,i);
}
int inf=dis[0][0][0];
fo(i,1,n)
fo(j,1,n)
fo(ii,1,3)
fo(jj,1,3)
if((i!=j)&&(i!=f[2][i][ii])&&(i!=f[1][j][jj])
&&(j!=f[2][i][ii])&&(j!=f[1][j][jj])&&(f[2][i][ii]!=f[1][j][jj])
&&(dis[2][i][f[2][i][ii]]<inf)&&(dis[1][i][j]<inf)&&(dis[1][j][f[1][j][jj]]<inf)
&&(ans[0]<dis[2][i][f[2][i][ii]]+dis[1][i][j]+dis[1][j][f[1][j][jj]])){
ans[0]=max(ans[0],dis[2][i][f[2][i][ii]]+dis[1][i][j]+dis[1][j][f[1][j][jj]]);
ans[1]=f[2][i][ii];ans[2]=i;ans[3]=j;ans[4]=f[1][j][jj];
}

printf("%d %d %d %d",ans[1],ans[2],ans[3],ans[4]);
return 0;
}