「美团 CodeM 复赛」城市网络-CSDN博客

题目链接

题意分析

首先 \([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;
}