CSAcademy Beta Round #5 Long Journey

时间:2021-09-27 01:57:28

题目链接:https://csacademy.com/contest/arhiva/#task/long_journey/

大意是有一张无向不带权的图,两个人同时从s点出发,分别前往a点和b点,且每个人应该走s到a和s到b的最短路,问他们可以一起走的最大距离是多少。

我一开始的想法是以s为源点bfs,做出所有点的前驱,然后判断a回到s和b回到s有多少点是共享的。WA了,后来一想,这么做确实是错的,因为很有可能a回到s的路是一条b不会走的路。然后变了下思路,直接分别以s,a和b为源点bfs,做出三个dis数组,枚举图中所有点,判断是否在s到a以及s到b的最短路上,如果是的话,则更新答案。

 #include <iostream>
#include <vector>
#include <algorithm>
#include <string>
#include <cstring>
#include <cstdio>
#include <math.h>
#include <queue>
#include <stack>
#include <map>
#include <cassert>
#include <set>
using namespace std; const int N=; struct Edge {
int to,next;
Edge() {}
Edge(int _to,int _next):to(_to),next(_next) {}
} edge[N<<];
int idx=,head[N];
inline void addedge(int u,int v) {
edge[++idx]=Edge(v,head[u]);
head[u]=idx;
}
bool vis[N]; void bfs(int s,int *dis) {
memset(vis,false,sizeof vis);
queue<int>que;
que.push(s);
vis[s]=true;
dis[s]=;
while (!que.empty()) {
int u=que.front();
que.pop();
for (int k=head[u];k;k=edge[k].next) {
int v=edge[k].to;
if (vis[v]) continue;
dis[v]=dis[u]+;
vis[v]=true;
que.push(v);
}
}
}
int dis[][N];
int main () {
int n,m;
while (scanf("%d %d",&n,&m)==) {
int s,a,b;
scanf("%d %d %d",&s,&a,&b);
idx=;memset(head,,sizeof head);
for (int i=;i<=m;i++) {
int u,v;
scanf("%d %d",&u,&v);
addedge(u,v);
addedge(v,u);
}
bfs(s,dis[]);
bfs(a,dis[]);
bfs(b,dis[]);
int da=dis[][a];
int db=dis[][b];
int ret=;
for (int i=;i<=n;i++) {
if (dis[][i]+dis[][i]==da&&dis[][i]+dis[][i]==db) {
ret=max(ret,dis[][i]);
}
}
cout<<ret<<endl;
}
return ;
}