Eyjafjalla 题解(dfs序+线段树+倍增)

题目链接

题目思路

这个题目好像有1w种写法。。

我写下我的思路，首先观察题目可以发现其实就是可以找到\(x\)的最大祖先使得这个值小于\(r\)

这样可以使用倍增实现

那么题目就可以转换为结点\(x\)中子数中有多少个节点的\(t\)值大于等于\(l\)

然后用离线+线段树+dfs序 实现即可

代码

#include<bits/stdc++.h>
#define fi first
#define se second
#define debug cout<<"I AM HERE"<<endl;
using namespace std;
typedef long long ll;
const int maxn=2e5+5,inf=0x3f3f3f3f,mod=1e9+7;
const double eps=1e-6;
int n,q,tot;
int t[maxn];
int in[maxn],out[maxn];
int dp[maxn][30];
int tree[maxn<<2];
int ans[maxn];
vector<int> g[maxn];
struct node{
    int opt,node,val,id;
}e[maxn];
void update(int node,int l,int r,int pos){
    if(l==r){
        tree[node]++;
        return ;
    }
    int mid=(l+r)/2;
    if(mid>=pos) update(node<<1,l,mid,pos);
    else update(node<<1|1,mid+1,r,pos);
    tree[node]=tree[node<<1]+tree[node<<1|1];
}
int query(int node,int L,int R,int l,int r){
    if(L<=l&&r<=R){
        return tree[node];
    }
    int mid=(l+r)/2,sum=0;
    if(mid>=L) sum+=query(node<<1,L,R,l,mid);
    if(mid<R)  sum+=query(node<<1|1,L,R,mid+1,r);
    return sum;
}
void dfs(int u,int v){
    in[u]=++tot;
    for(auto i:g[u]){
        if(i==v) continue;
        dp[i][0]=u;
        dfs(i,u);
    }
    out[u]=tot;
}
bool cmp(node a,node b){
    if(a.val!=b.val){
        return a.val>b.val;
    }else{
        return a.opt<b.opt;
    }
}
signed main(){
    scanf("%d",&n);
    for(int i=1,u,v;i<=n-1;i++){
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(1,1);
    for(int j=1;j<=20;j++){//第一层枚举长度
        for(int i=1;i<=n;i++){
            dp[i][j]=dp[dp[i][j-1]][j-1];
        }
    }
    int cnt=0;
    for(int i=1;i<=n;i++){
        scanf("%d",&t[i]);
        e[++cnt]={1,i,t[i],-1};
    }
    scanf("%d",&q);
    for(int i=1,id,l,r;i<=q;i++){
        scanf("%d%d%d",&id,&l,&r);
        if(t[id]>r||t[id]<l) continue;
        for(int j=20;j>=0;j--){//
            if(t[dp[id][j]]!=0&&t[dp[id][j]]<=r){
                id=dp[id][j];
            }
        }
        e[++cnt]={2,id,l,i};
    }
    sort(e+1,e+1+cnt,cmp);
    for(int i=1;i<=cnt;i++){
        if(e[i].opt==1){
            update(1,1,n,in[e[i].node]);
        }else{
            ans[e[i].id]=query(1,in[e[i].node],out[e[i].node],1,n);
        }
    }
    for(int i=1;i<=q;i++){
        printf("%d\n",ans[i]);
    }
    return 0;
}