桂林电子科技大学第三届ACM程序设计竞赛 D 寻找

链接：https://ac.nowcoder.com/acm/contest/558/D
来源：牛客网
 

小猫在研究树。

小猫在研究树上的距离。

给定一棵N个点的树，每条边边权为1。

Q次询问，每次给定a,b,c，请你输出a到b的路径上离c最近的点的编号。

输入描述:

第一行一个正整数N，表示节点数量。

接下来N−1行，第i行两个正整数ui,vi，表示第i条边连接节点ui,vi。

接下来一行一个正整数Q，表示询问数量。

接下来Q行，每行三个正整数a,b,c，表示一组询问。

输出描述:

Q行，每行一个正整数，表示每个询问的答案。

示例1

输入

复制

5
1 2
1 3
2 4
2 5
3
1 2 3
4 5 1
1 4 5

输出

复制

1
2
2

 看完题之后画了个图，感觉是个LCA的题

首先找到  a b c 两两之间的lca

大致分为两种情况：

1、如果ac  ，bc之间的最近公共祖先的深度最小值   小于 ab之间最近公共祖先的深度 

那么明显答案就是ab的最近公共祖先

2、反之，就输出与c点距离最近的，判断一下深度就行了  ans要么是ac的最近公共祖先，要么是bc的最近公共祖先

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5;
int dep[N],Fa[N][22],n,q;
vector<int>G[N];
void dfs(int u,int fa)
{
    dep[u]=dep[fa]+1;
    Fa[u][0]=fa;
    for(int i=1;(1<<i)<=dep[u];i++) Fa[u][i]=Fa[Fa[u][i-1]][i-1];
    for(auto &v:G[u])
    {
        if(v==fa) continue;
        dfs(v,u);
    }
}
int lca(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=20;i>=0;i--) if((1<<i)<=dep[x]-dep[y]) x=Fa[x][i];
    if(x==y) return x;
    for(int i=20;i>=0;i--) if(Fa[x][i]!=Fa[y][i]) x=Fa[x][i],y=Fa[y][i];
    return Fa[x][0];
}
int dis(int x,int y)
{ 
	return dep[x]+dep[y]-2*dep[lca(x,y)];
}
int main(int argc, char const *argv[])
{
	scanf("%d",&n);
	for(int i=0;i<n-1;i++)
	{
		int u,v;scanf("%d%d",&u,&v);
		G[u].emplace_back(v);
		G[v].emplace_back(u);
	}
	dfs(1,0);
	scanf("%d",&q);
	while(q--)
	{
		int a,b,c;scanf("%d%d%d",&a,&b,&c);
		int ab=lca(a,b);
		int ac=lca(a,c);
		int bc=lca(b,c); int mi=1e9+7;
		mi=min(dep[ac],dep[bc]);
		if(mi<dep[ab]) printf("%d\n", ab);
		else printf("%d\n",dep[c]-dep[ac]<dep[c]-dep[bc] ? ac:bc);
	}
	return 0;
}