[BZOJ4756][Usaco2017 Jan]Promotion Counting（可持久化线段树）

题目描述

传送门

题解

主席树好久没打也是各种手残啊
这题也是线段树合并的裸题过几天写一发！

代码

#include
#include
#include
#include
#include
using namespace std;
#define N 100005

int n,LSH,sz,dfs_clock,a[N],lsh[N];
int tot,point[N],nxt[N],v[N];
int in[N],out[N],root[N],sum[N*30],ls[N*30],rs[N*30];

void add(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
}
void update(int &now,int l,int r,int x)
{
    int mid=(l+r)>>1;
    sum[++sz]=sum[now]+1,ls[sz]=ls[now],rs[sz]=rs[now];
    now=sz;
    if (l==r) return;
    if (x<=mid) update(ls[now],l,mid,x);
    else update(rs[now],mid+1,r,x);
}
int query(int L,int R,int l,int r,int lr,int rr)
{
    int mid=(l+r)>>1,ans=0;
    if (lr<=l&&r<=rr) return sum[R]-sum[L];
    if (lr<=mid) ans+=query(ls[L],ls[R],l,mid,lr,rr);
    if (mid+1<=rr) ans+=query(rs[L],rs[R],mid+1,r,lr,rr);
    return ans;
}
void dfs(int x)
{
    in[x]=++dfs_clock;
    root[in[x]]=root[in[x]-1];
    update(root[in[x]],1,LSH,a[x]);
    for (int i=point[x];i;i=nxt[i])
        dfs(v[i]);
    out[x]=dfs_clock;
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);
        lsh[++LSH]=a[i];
    }
    sort(lsh+1,lsh+LSH+1);LSH=unique(lsh+1,lsh+LSH+1)-lsh-1;
    for (int i=1;i<=n;++i) a[i]=lower_bound(lsh+1,lsh+LSH+1,a[i])-lsh;
    for (int i=2;i<=n;++i)
    {
        int fa;scanf("%d",&fa);
        add(fa,i);
    }
    dfs(1);
    for (int i=1;i<=n;++i)
    {
        int ans=query(root[in[i]-1],root[out[i]],1,LSH,a[i]+1,LSH);
        printf("%d\n",ans);
    }
}