bzoj4756线段树合并模板

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 100003
using namespace std;
int n,a[N],b[N],tot;
int head[N];
int tr[N*20],ls[N*20],rs[N*20],root[N],cnt,sz,ans[N];
struct edge
{
    int v,next;
}edges[N*2];
void init()
{
    memset(head,-1,sizeof(head));
    tot=0;
}
void add(int x,int y)
{
   edges[tot].next=head[x];
   edges[tot].v=y;
   head[x]=tot++;
}
void update(int now)
{
    tr[now]=tr[ls[now]]+tr[rs[now]];
}
void insert(int &i,int l,int r,int x)
{
    if (l>r) return;
    i=++sz;
    if (l==r) {
        tr[i]=1;
        return;
    }
    int mid=(l+r)/2;
    if (x<=mid) insert(ls[i],l,mid,x);
    else insert(rs[i],mid+1,r,x);
    update(i);
}
int merge(int x,int y)
{
    if (!x) return y;
    if (!y) return x;
    ls[x]=merge(ls[x],ls[y]);
    rs[x]=merge(rs[x],rs[y]);
    update(x);
    return x;
}
int qjsum(int now,int l,int r,int ll,int rr)
{
    if (rr<ll) return 0;
    if (ll<=l&&r<=rr) return tr[now];
    int mid=(l+r)/2; int ans=0;
    if (ll<=mid) ans+=qjsum(ls[now],l,mid,ll,rr);
    if (rr>mid) ans+=qjsum(rs[now],mid+1,r,ll,rr);
    return ans;
}
void solve(int x,int fa)
{
    for (int i=head[x];i!=-1;i=edges[i].next){
    int v=edges[i].v;
    if(v==fa)
        continue;
        solve(v,x);
     root[x]=merge(root[x],root[v]);
    }
    ans[x]=qjsum(root[x],1,n,a[x]+1,cnt);
}
int main()
{
 init();
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
     scanf("%d",&a[i]),b[i]=a[i];
    sort(b+1,b+n+1);
    cnt=unique(b+1,b+n+1)-b-1;
    for (int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+cnt+1,a[i])-b;
    for (int i=2;i<=n;i++) {
        int x; scanf("%d",&x);
        add(x,i);
    }
    for (int i=1;i<=n;i++) insert(root[i],1,n,a[i]);
    solve(1,-1);
    for (int i=1;i<=n;i++) printf("%d\n",ans[i]);
}

 

来源：https://blog.csdn.net/sdauguanweihong/article/details/100882295