【洛谷P3605】晋升者计数

题目大意：给定一棵 N 个点的树，点有点权，求对于每个点来说，以该点为根的子树内比该点点权小的点的个数。

题解：考虑对于每个点开一棵权值线段树。递归过程中，将子树的信息合并到父节点上，统计答案后，再将父节点信息加入权值线段树即可。

代码如下

#include <bits/stdc++.h>
#define pb push_back
using namespace std;
const int maxn=1e5+10;
typedef long long ll;

int n,a[maxn],d[maxn],cnt,ans[maxn];
vector<int> G[maxn];
struct node{
    #define ls(o) t[o].lc
    #define rs(o) t[o].rc
    int lc,rc,sz;
}t[maxn*20];
int tot,root[maxn];
inline void pushup(int o){t[o].sz=t[ls(o)].sz+t[rs(o)].sz;}
void insert(int &o,int l,int r,int pos){
    if(!o)o=++tot;
    if(l==r){++t[o].sz;return;}
    int mid=l+r>>1;
    if(pos<=mid)insert(ls(o),l,mid,pos);
    else insert(rs(o),mid+1,r,pos);
    pushup(o);
}
int query(int o,int l,int r,int x,int y){
    if(!o)return 0;
    if(l==x&&r==y)return t[o].sz;
    int mid=l+r>>1;
    if(y<=mid)return query(ls(o),l,mid,x,y);
    else if(x>mid)return query(rs(o),mid+1,r,x,y);
    else return query(ls(o),l,mid,x,mid)+query(rs(o),mid+1,r,mid+1,y);
}
int merge(int x,int y,int l,int r){
    if(!x||!y)return x+y;
    if(l==r){t[x].sz+=t[y].sz;return x;}
    int mid=l+r>>1;
    ls(x)=merge(ls(x),ls(y),l,mid);
	rs(x)=merge(rs(x),rs(y),mid+1,r);
    return pushup(x),x;
}

void read_and_parse(){// 1 - n
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]),d[i]=a[i];
    sort(d+1,d+n+1);
    cnt=unique(d+1,d+n+1)-d-1;
    for(int i=1;i<=n;i++)a[i]=lower_bound(d+1,d+cnt+1,a[i])-d;
    for(int i=2,fa;i<=n;i++)scanf("%d",&fa),G[fa].pb(i);
}

void dfs(int u){
    for(auto v:G[u]){
        dfs(v);
        root[u]=merge(root[u],root[v],1,n);
    }
    ans[u]=t[root[u]].sz-query(root[u],1,n,1,a[u]-1);
    insert(root[u],1,n,a[u]);
}

void solve(){
    dfs(1);
    for(int i=1;i<=n;i++)printf("%d\n",ans[i]);
}

int main(){
    read_and_parse();
    solve();
    return 0;
}