「美团 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;
}
