luogu P3605 [USACO17JAN]Promotion Counting晋升者计数 BZOJ 4756

题目链接1
题目链接2

题意

给出每个点的父亲节点，求每个点的孩子节点有多少个权值大于它。

题解

线段树合并模板题啦，从下往上合并权值线段树即可（略略略

//Suplex
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#define N 100100
using namespace std;
int n,tot,num,cnt,fa[N],ans[N],vet[N+N],Next[N+N],head[N],t[2000000],ls[2000000],rs[2000000],root[N],q[N];

struct leo{
    int val,id;
}a[N];

bool cmp(leo a,leo b)
{
    return a.val<b.val;
}

void add_edge(int u,int v)
{
    vet[++cnt]=v;Next[cnt]=head[u];head[u]=cnt;
}

inline void modify(int &rt,int l,int r,int x)
{
    if(!rt) rt=++num;
    t[rt]++;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(x<=mid) modify(ls[rt],l,mid,x);
    else modify(rs[rt],mid+1,r,x);
}

inline int query(int rt,int l,int r,int x)
{
    if(!rt) return 0;
    if(l>=x) return t[rt];
    int mid=(l+r)>>1;
    if(x<=mid) return query(ls[rt],l,mid,x)+query(rs[rt],mid+1,r,x);
    return query(rs[rt],mid+1,r,x);
}

inline int merge(int u,int v)
{
    if(!u) return v;
    if(!v) return u;
    int xx=++num;
    t[xx]=t[u]+t[v];
    ls[xx]=merge(ls[u],ls[v]);
    rs[xx]=merge(rs[u],rs[v]);
    return xx;
}

void dfs(int u)
{
    for(int i=head[u];i;i=Next[i]){
        int v=vet[i];
        if(v==fa[u]) continue;
        dfs(v);
        root[u]=merge(root[u],root[v]);
    }
    ans[u]=query(root[u],1,tot,q[u]+1);
    modify(root[u],1,tot,q[u]);
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i].val),a[i].id=i;
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=n;i++){
        if(a[i].val != a[i-1].val) tot++;
        q[a[i].id]=tot;
    }
    for(int i=2;i<=n;i++){
        scanf("%d",&fa[i]);
        add_edge(i,fa[i]);add_edge(fa[i],i);
    }
    dfs(1);
    for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
    return 0;
}