zzuli 2520: 大小接近的点对（离散化+线段树合并）

题意：

给定一棵n个点的树，点1为根，第i个点的点权为a(i)。
给定整数K，如果u是v的祖先，且abs(a[u]−a[v])≤K那么(u,v)就是“大小接近的点对 ”。
对于树上的每个结点 i，你都需要计算以其为根的子树中的“大小接近的点对”的数量。

数据范围：n<=1e5，k<=1e9，a(i)<=1e9

解法：

显然需要将a[i],a[i]-K,a[i]+k,加在一起离散化,
dfs整棵树,每棵树开一颗动态开点权值线段树,
ans[x]+=ans[v],然后将v的树合并到x上,在rt[x]上查询有多少个范围在[a[i]-K,a[i]+K]内的点.

ps:
也可以用树状数组做,每次dfs之前记录有多少个范围在[a[i]-K,a[i]+K]的点,记为cnt1,
dfs出来之后记录有多少个范围在[a[i]-K,a[i]+K]的点,记为cnt2,
那么ans[x]+=cnt2-cnt1.

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=3e5+5;
vector<int>g[maxm];
int x[maxm],num;
int ans[maxm];
int L[maxm];
int R[maxm];
int a[maxm];
int n,K;
//
int lc[maxm*10],rc[maxm*10],cnt[maxm*10];
int rt[maxm],tot;
void add(int x,int val,int l,int r,int &k){
    if(!k)k=++tot;
    cnt[k]++;
    if(l==r)return ;
    int mid=(l+r)/2;
    if(x<=mid)add(x,val,l,mid,lc[k]);
    else add(x,val,mid+1,r,rc[k]);
}
int emerged(int x,int y){
    if(!x||!y)return x+y;
    lc[x]=emerged(lc[x],lc[y]);
    rc[x]=emerged(rc[x],rc[y]);
    cnt[x]+=cnt[y];
    return x;
}
int ask(int st,int ed,int l,int r,int k){
    if(!k||!cnt[k])return 0;
    if(st<=l&&ed>=r)return cnt[k];
    int mid=(l+r)/2;
    int ans=0;
    if(st<=mid)ans+=ask(st,ed,l,mid,lc[k]);
    if(ed>mid)ans+=ask(st,ed,mid+1,r,rc[k]);
    return ans;
}
void dfs(int x){
    for(int v:g[x]){
        dfs(v);
        ans[x]+=ans[v];
        rt[x]=emerged(rt[x],rt[v]);
    }
    add(a[x],1,1,num,rt[x]);
    ans[x]+=ask(L[x],R[x],1,num,rt[x]);
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    //init
    cin>>n>>K;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=2;i<=n;i++){
        int fa;cin>>fa;
        g[fa].push_back(i);
    }
    //离散化
    for(int i=1;i<=n;i++){
        x[++num]=a[i];
        x[++num]=a[i]+K;
        x[++num]=a[i]-K;
    }
    sort(x+1,x+1+num);
    num=unique(x+1,x+1+num)-x-1;
    for(int i=1;i<=n;i++){
        L[i]=lower_bound(x+1,x+1+num,a[i]-K)-x;
        R[i]=lower_bound(x+1,x+1+num,a[i]+K)-x;
        a[i]=lower_bound(x+1,x+1+num,a[i])-x;
    }
    dfs(1);
    for(int i=1;i<=n;i++){
        cout<<ans[i]<<'\n';
    }
    return 0;
}

---