线段树合并

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
using namespace std;
int n,m,t,tot;
int first[N],v[N],nxt[N];
int a[N],ans[N],val[N],root[N],size[N*40],lc[N*40],rc[N*40];
void add(int x,int y)
{
    t++;
    nxt[t]=first[x];
    first[x]=t;
    v[t]=y;
}
void build(int &root,int l,int r,int val)
{
    root=++tot;
    size[root]=1;
    if(l==r)  return;
    int mid=(l+r)>>1;
    if(val<=mid)  build(lc[root],l,mid,val);
    else  build(rc[root],mid+1,r,val);
}
int Merge(int x,int y)
{
    if(!x)  return y;
    if(!y)  return x;
    size[x]+=size[y];
    lc[x]=Merge(lc[x],lc[y]);
    rc[x]=Merge(rc[x],rc[y]);
    return x;
}
int calc(int root,int l,int r,int x,int y)
{
    if(l>=x&&r<=y)
      return size[root];
    int ans=0,mid=(l+r)>>1;
    if(x<=mid)  ans+=calc(lc[root],l,mid,x,y);
    if(y>mid)  ans+=calc(rc[root],mid+1,r,x,y);
    return ans;
}
void dfs(int x)
{
    int i,j;
    for(i=first[x];i;i=nxt[i])
    {
        j=v[i],dfs(j);
        root[x]=Merge(root[x],root[j]);
    }
    ans[x]=calc(root[x],1,m,val[x]+1,m);
}
int main()
{
    int x,i;
    scanf("%d",&n);
    for(i=1;i<=n;++i)  scanf("%d",&a[i]),val[i]=a[i];
    for(i=2;i<=n;++i)  scanf("%d",&x),add(x,i);
    sort(a+1,a+n+1),m=unique(a+1,a+n+1)-(a+1);
    for(i=1;i<=n;++i)
    {
        val[i]=lower_bound(a+1,a+m+1,val[i])-a;
        build(root[i],1,m,val[i]);
    }
    dfs(1);
    for(i=1;i<=n;++i)
      printf("%d\n",ans[i]);
    return 0;
}