bzoj 4756 Promotion Counting —— 线段树合并

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=4756

合并子树的权值线段树；

merge 返回 int 或者是 void 都可以。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mid ((l+r)>>1)
using namespace std;
int const xn=1e5+,xm=xn*;//
int n,p[xn],b[xn],hd[xn],ct,to[xn<<],nxt[xn<<];
int cnt,rt[xn],ls[xm],rs[xm],sum[xm],ans[xn],tot;
int rd()
{
  int ret=,f=; char ch=getchar();
  while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}
  while(ch>=''&&ch<='')ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
  return f?ret:-ret;
}
int gt[];
void wr(int x)
{
  if(!x){puts(""); return;}
  if(x<)putchar('-'),x=-x;
  int t=;
  while(x)gt[++t]=x%,x/=;
  for(int i=t;i;i--)putchar(gt[i]+'');
  puts("");
}
void add(int x,int y){to[++ct]=y; nxt[ct]=hd[x]; hd[x]=ct;}
void merge(int &x,int y)
{
  if(!x){x=y; return;}
  sum[x]+=sum[y];
  if(ls[y])merge(ls[x],ls[y]);//if
  if(rs[y])merge(rs[x],rs[y]);//if
}
/*
int merge(int x,int y,int l,int r)
{
  if(!x||!y)return x+y;
  sum[x]+=sum[y];
  if(l==r)return x;
  ls[x]=merge(ls[x],ls[y],l,mid);
  rs[x]=merge(rs[x],rs[y],mid+1,r);
  return x;
}
*/
void add(int &x,int l,int r,int pos)
{
  if(!x)x=++cnt; sum[x]++;
  if(l==r)return;
  if(pos<=mid)add(ls[x],l,mid,pos);
  else add(rs[x],mid+,r,pos);
}
int query(int x,int l,int r,int L,int R)
{
  if(!x)return ;
  if(l>=L&&r<=R)return sum[x];
  int ret=;
  if(mid>=L)ret+=query(ls[x],l,mid,L,R);
  if(mid<R)ret+=query(rs[x],mid+,r,L,R);
  return ret;
}
void dfs(int x)
{
  for(int i=hd[x],u;i;i=nxt[i])
    {
      dfs(u=to[i]);
      //      rt[x]=merge(rt[x],rt[u],1,tot);
      merge(rt[x],rt[u]);
    }
  ans[x]=query(rt[x],,tot,p[x],tot);
  add(rt[x],,tot,p[x]);
}
int main()
{
  n=rd();
  for(int i=;i<=n;i++)p[i]=rd(),b[i]=p[i];
  sort(b+,b+n+); tot=unique(b+,b+n+)-b-;
  for(int i=;i<=n;i++)p[i]=lower_bound(b+,b+tot+,p[i])-b;
  for(int i=,fa;i<=n;i++)fa=rd(),add(fa,i);
  dfs();
  for(int i=;i<=n;i++)wr(ans[i]);
  return ;
}