题目描述
每年的BNU校赛都会有两次赛前培训,为此就需要去借教室,由于SK同学忙于出题,这个事情就由小Q同学来跑腿。SK同学准备从宿舍出发,把借教室的单子交给小Q同学让他拿去教务处盖章,但是何老师突然发现SK同学好像借错教室了,想抢在借教室的单子被送到教务处之前拦截下来。
现在把校园抽象成一棵n个节点的树,每条边的长度都是一个单位长度,从1到n编号,其中教务处位于1号节点,接下来有q个询问,每次询问中SK同学会从B号节点出发,到C号节点找到小Q同学并将借教室的单子交给他,然后小Q同学再从C号节点出发前往教务处,何老师会从A号节点出发开始拦截。
所有人在一个单位时间内最多走一个单位距离,只要何老师在单子还没被送到教务处之前遇到拿着单子的同学都算拦截成功,如果小Q同学已经到了教务处,那么由于小Q同学手速极快,单子会被立即交上去,即使何老师到了教务处也无济于事,你需要判断何老师是否能够拦截成功。
输入描述:
第一行是一个正整数T(≤ 5),表示测试数据的组数, 对于每组测试数据, 第一行是两个整数n,q(1≤ n,q ≤ 100000),分别表示节点数和询问数, 接下来n-1行,每行包含两个整数x,y(1≤ x,y ≤ n),表示x和y之间有一条边相连,保证这些边能构成一棵树, 接下来q行,每行包含三个整数A,B,C(1 ≤ A,B,C ≤ n),表示一个询问,其中A是何老师所在位置,B是SK同学所在位置,C是小Q同学所在位置,保证小Q同学初始不在教务处。
输出描述:
对于每个询问,输出一行,如果何老师能成功拦截则输出"YES"(不含引号),否则输出"NO"(不含引号)。
示例1
输入
1
7 2
1 2
2 3
3 4
4 7
1 5
1 6
3 5 6
7 5 6
输出
YES
NO
最近公共祖先
bfs版本
#include<bits/stdc++.h>
using namespace std;
struct node
{
int v,w;
node(int a,int b){
v=a;
w=b;
}
};
vector<node>G[100100];
const int N=1e5+100;
int depth[N],fa[N][20];
void bfs()
{
memset(depth,0x3f,sizeof depth);
depth[0]=0;depth[1]=1;
queue<int> q;
q.push(1);
while(q.size())
{
int t=q.front();
q.pop();
for(int i=0;i<G[t].size();i++)
{
int v=G[t][i].v;
if(depth[v]>depth[t]+1)
{
depth[v]=depth[t]+1;
q.push(v);
fa[v][0]=t;
for(int k=1;k<=18;k++) fa[v][k]=fa[fa[v][k-1]][k-1];
}
}
}
}
int lca(int x,int y)
{
if(depth[x]<depth[y]) swap(x,y);
for(int k=18;k>=0;k--) if(depth[fa[x][k]]>=depth[y]) x=fa[x][k];
if(x==y) return x;
for(int k=18;k>=0;k--)
{
if(fa[x][k]!=fa[y][k])
{
x=fa[x][k];
y=fa[y][k];
}
}
return fa[x][0];
}
int getdis(int x,int y)
{
return depth[x]+depth[y]-2*depth[lca(x,y)];
}
int main(){
int n;
cin>>n;
while(n--){
int t,q;
cin>>t>>q;
memset(fa,0,sizeof fa);
for(int i=1;i<=t;i++) G[i].clear();
for(int i=1;i<t;i++){
int x,y;cin>>x>>y;
G[x].push_back(node(y,1));
G[y].push_back(node(x,1));
}
bfs();
while(q--)
{
int a,b,c;
scanf("%d %d %d",&a,&b,&c);
int a1=getdis(a,1);
int bc=getdis(c,b);
int c1=getdis(c,1);
int ac=getdis(a,c);
if(a1<bc+c1)
cout<<"YES"<<endl;
else if(ac<=bc) cout<<"YES"<<endl;
else if(lca(a,c)!=1&&a1<=bc+c1)
cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
}
}
dfs版本
void dfs(int u){
for(int i=0;i<G[u].size();i++){
int v=G[u][i];
if(v!=fa[u][0]){
fa[v][0]=u,depth[v]=depth[u]+1;
for(int i=1;i<=mx;i++)
fa[v][i]=fa[fa[v][i-1]][i-1];
dfs(v);
}
}
}
mx=log2(n);depth[1]=0;