「美团 CodeM 复赛」城市网络

题目链接

题意分析

首先 \([u,v]\)在树上是一条深度递增的链

那么我们可以使用倍增找 \(x\)的祖先当中深度最大的值大于\(x\)的点

然后维护一个\(pre\)

重新建树

这样从\(x\)到根节点 权值严格递增

我们在这个新树上维护好深度以及倍增数组

然后再去倍增找到最近的权值>z的点

如果不存在或者深度\(<y\) 那么输出\(0\)

然后再倍增找到找到深度大于\(y\)的最浅的节点

然后用深度就可以计算出节点的个数

CODE:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<string>
#include<queue>
#include<map>
#include<stack>
#include<list>
#include<set>
#include<deque>
#include<vector>
#include<ctime>
#define ll long long
#define inf 0x7fffffff
#define N 500008
#define IL inline
#define M 100611
#define D double
#define ull unsigned long long
#define R register
using namespace std;
template<typename T>IL void read(T &_)
{
    T __=0,___=1;char ____=getchar();
    while(!isdigit(____)) {if(____=='-') ___=0;____=getchar();}
    while(isdigit(____)) {__=(__<<1)+(__<<3)+____-'0';____=getchar();}
    _=___ ? __:-__;
}
/*-------------OIʹ�ҿ���-------------*/
int n,q,tot;
int to[M<<1],nex[M<<1],head[M],fa[M][20],dis[M][20],fath[M][20];
int num[M],in[M],deep[M],dep[M];
IL void add(int x,int y)
{to[++tot]=y;nex[tot]=head[x];head[x]=tot;in[y]++;}
IL void dfs(int now,int fat)
{
	dep[now]=dep[fat]+1;fa[now][0]=fat;dis[now][0]=num[now];
	for(R int i=1;(1<<i)<=dep[now];++i)
	{
		fa[now][i]=fa[fa[now][i-1]][i-1];
		dis[now][i]=max(dis[now][i-1],dis[fa[now][i-1]][i-1]);
	}
	for(R int i=head[now];i;i=nex[i])
	{
		int v=to[i];
		if(v==fat) continue;
		dfs(v,now);
	}
}
IL void DFS(int now,int fat)
{
	fath[now][0]=fat;deep[now]=deep[fat]+1;
	for(R int i=1;(1<<i)<=deep[now];++i)
	fath[now][i]=fath[fath[now][i-1]][i-1];
	for(R int i=head[now];i;i=nex[i])
	{
		int v=to[i];
		if(v==fat) continue;
		DFS(v,now);
	}
}
int main()
{
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);
	read(n);read(q);
	for(R int i=1;i<=n;++i) read(num[i]);
	for(R int i=1,x,y;i<n;++i)
	{
		read(x);read(y);
		add(x,y);add(y,x);
	}
	dfs(1,0);
	tot=0;
	memset(in,0,sizeof in);
	memset(head,0,sizeof head);
	for(R int i=2;i<=n;++i)
	{
		int now=i;
		for(R int j=19;j>=0;--j)
		if((1<<j)<=dep[now]&&dis[now][j]<=num[i]) now=fa[now][j];
		if(num[now]>num[i]) add(now,i);
	}
	for(R int i=1;i<=n;++i)
	if(!in[i]) DFS(i,0);
 	while(q--)
 	{
 		int x,y,z;read(x);read(y);read(z);int now=x;
		for(R int i=19;i>=0;--i)
		if((1<<i)<=dep[now]&&dis[now][i]<=z) now=fa[now][i];

		if(num[now]<=z||dep[y]>dep[now]) puts("0");
		else
		{
			int res=deep[now];
			for(R int i=19;i>=0;--i)
			if((1<<i)<=deep[now]&&dep[fath[now][i]]>=dep[y]) now=fath[now][i];
			res-=deep[now];res++;
			printf("%d\n",res);
		}
	}
//	fclose(stdin);
//	fclose(stdout);
    return 0;
}

HEOI 2019 RP++