洛谷 P6374 树上询问

$$洛谷P6374树上询问$$

题目描述 ：洛谷 P6374

思路：

1. 暴力枚举每个点$i$，判断$LCA(a,b)$是否=$c$。复杂度$O(n\cdot q\cdot lo{g}_2(n))$，用脚想想都知道过不了（能骗一分是一分）
2. 貌似可以用那几个特殊数据小搞一波，估计比暴力多个10分吧。（一分也是分）
3. 说说正解：
   首先，我们假设结点1为树根，敲一遍$DFS$和$LCA$的预处理
   设dp[x]为以x为根的子树的节点数。
   总共有四种情况：
   1.$c$不在$a=>b$的路径上，此时答案为0.
   2.$c$是$a,b$的$LCA$，此时答案为c与其祖先们的节点。

   重点来袭

   3. c为a~LCA(a,b)路径上的一点，此时$answer$=dp[c]-dp[v] v表示v为a所在子树的根节点（以c为树根时）
   4. c为b~LCA(a,b)路径上的一点，此时$answer$=dp[c]-dp[v] v表示v为b所在子树的根节点（以c为树根时）

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#define r register
#define rep(i,x,y) for(r int i=x;i<=y;++i)
#define per(i,x,y) for(r int i=x;i>=y;--i)
using namespace std;
typedef long long ll;
const int V=5*1e5+100;
ll n,q,x,y,z,top;
ll f[V][23],d[V],dp[V],head[V];
struct node
{
	ll next,to;
}e[V<<1];
void add(ll x,ll y)
{
	e[++top]=node{head[x],y};
	head[x]=top;
}
void dfs(ll x,ll y)
{
	d[x]=d[y]+1;
	dp[x]=1;
	f[x][0]=y;
	for(r ll i=head[x];i;i=e[i].next)
	{
		ll v=e[i].to;
		if(v!=y)
		{
			dfs(v,x);
			dp[x]+=dp[v]; //子树大小
		}
	}
}
ll LCA(ll x,ll y) //倍增求LCA
{
	if(d[x]>d[y]) swap(x,y);
	ll len=d[y]-d[x],k=20,t=1<<k;
	while(len)
	{
		if(len>=t)
		 len-=t,y=f[y][k];
		t>>=1,--k;
	}
	if(x==y) return x;
	per(k,20,0)
	 if(f[x][k]!=f[y][k])
	 {
	 	x=f[x][k];
	 	y=f[y][k];
	 }
	return f[x][0];
}
bool check(ll x,ll y, ll z) //判断y是否在x-->z的路径上
{
	ll k=LCA(x,z);
	return ((LCA(x,y)==y)||(LCA(y, z)==y))&&LCA(y,k)==k;
}
ll size(ll x,ll y) //找到点“v”
{
	if(x==y) return 0;
	ll len=d[x]-d[y],k=20,t=1<<k;
	per(i,20,0)
	 if(d[f[x][i]]>d[y])
	  x=f[x][i];
	return dp[x]; 
}
int main()	
{
	scanf("%lld%lld",&n,&q);
	rep(i,1,n-1)
	{
		scanf("%lld%lld",&x,&y);
		add(x,y),add(y,x);
	}
	dfs(1,0);
	rep(j,1,20)
	 rep(i,1,n)
	  f[i][j]=f[f[i][j-1]][j-1]; //LCA的预处理
	rep(i,1,q)
	{
		scanf("%lld%lld%lld",&x,&y,&z);
		ll v=LCA(x,y);
		if(v==z) printf("%lld\n",n-size(x,v)-size(y,v));
		else if(check(x,z,v)) printf("%lld\n",dp[z]-size(x,z));
		else if(check(y,z,v)) printf("%lld\n",dp[z]-size(y,z));
		else printf("0\n");
	}
	return 0;
}