2021牛客暑期多校训练营9 E Eyjafjalla

题目意思

给出一个树，每个节点有一个温度，并且规定父亲结点温度一定大于儿子节点。
给出一个温度区间[l,r]和操作的点x，如果点x温度在该区间中，那么将该点加入集合，并且判断x的相邻结点是否满足该条件，问最终集合中元素能有多少个。

题目思路

由于题目条件非常明确的告诉了我们祖先节点温度会大于子孙节点，我们发现对于一个点y向下找满足条件的点时，只需要找他的子树上有多少个点的温度大于l就好了。
那么我们对于每次询问可以先找到x的深度最小的温度小于r的祖先节点a，然后我们只要找出a结点的子树上有多少个点温度是大于l的就行了。
查询子树的操作，我们用dfs序将每个位置存在线段树中，线段树维护一个区间最大值一个区间最小值。每次查询的子树就是它入栈到出栈的区间，判断区间是否符合就行了。
找祖先的操作是用倍增方法求解的

ac代码

 #include <stdio.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <utility>
#include<unordered_map>
#define pi 3.1415926535898
#define ll long long
#define ull unsigned long long
#define lson rt<<1
#define rson rt<<1|1
#define eps 1e-6
#define ms(a,b) memset(a,b,sizeof(a))
#define legal(a,b) a&b
#define lowbit(x) (x & (-x))
#define print1 printf("111\n")
#define endl "\n"
#define pl printf("\n")
#define pb(x) push_back(x)
#define pii pair<int,int>
#define pair4 pair<pair<int,int>,pair<int,int> >
#define fi first
#define se second
using namespace std;
const int maxn = 3e6+10;
const int inf = 1e9+10;
const ll llinf =1e18+10;
const ll mod = 998244353;
const int p=31;

int n,tim;
int in[maxn],pos[maxn],out[maxn],a[maxn];
int dp[maxn][30];
pii t[maxn];
vector<int>vec[maxn];

void dfs(int x,int fa)
{
    in[x]=++tim;
    dp[x][0]=fa;
    for(int i=1;i<=20;i++)dp[x][i]=dp[dp[x][i-1]][i-1];
    for(auto to:vec[x])
    {
        if(to==fa)continue;
        dfs(to,x);
    }
    out[x]=tim;
}

void build(int rt,int l,int r)
{
    if(l==r)
    {
        t[rt].fi=a[l];
        t[rt].se=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(lson,l,mid);
    build(rson,mid+1,r);
    t[rt].fi=max(t[lson].fi,t[rson].fi);
    t[rt].se=min(t[lson].se,t[rson].se);
}

int query(int rt,int l,int r,int L,int R,int mi)
{
    if(L<=l&&r<=R)
    {
        if(t[rt].se>=mi)
            return r-l+1;
        if(t[rt].fi<mi)
            return 0;
    }
    int mid=(l+r)>>1;
    int ans=0;
    if(L<=mid)
        ans+=query(lson,l,mid,L,R,mi);
    if(R>mid)
        ans+=query(rson,mid+1,r,L,R,mi);
    return ans;
}

int main()
{
    scanf("%d",&n);
    for(int i=1,u,v;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        vec[u].push_back(v);
        vec[v].push_back(u);
    }
    dfs(1,1);
    for(int i=1,x;i<=n;i++)
    {
        scanf("%d",&x);
        a[in[i]]=x;
    }
    build(1,1,n);
    int q;
    scanf("%d",&q);
    for(int i=1,op,l,r;i<=q;i++)
    {
        scanf("%d%d%d",&op,&l,&r);

        if(a[in[op]]>r||a[in[op]]<l)
        {
            printf("0\n");
            continue;
        }

        for(int i=20;i>=0;i--)
        {
            if(a[in[dp[op][i]]]>r)continue;
            op=dp[op][i];
        }
        //print1;
        //printf("%d\n",op);
        printf("%d\n",query(1,1,n,in[op],out[op],l));
    }

}