[COGS 1535] [ZJOI2004]树的果实 树状数组+桶

我们用树状数组做差就可以解决一切问题，我用桶排并用此来表示出第几大就可以直接求前缀和了

#include<cstdio>
#include<algorithm>
#define MAXN 100010
using namespace std;
inline int read()
{
   int sum=;
   char ch=getchar();
   while(ch<''||ch>'')ch=getchar();
   while(ch>=''&&ch<='')
   {
     sum=(sum<<)+(sum<<)+ch-'';
     ch=getchar();
   }
   return sum;
}
int P[MAXN],H[MAXN];
int n,m;
struct VIA
{
  int to,next;
}c[MAXN];
int head[MAXN],t;
inline void add(int x,int y)
{
  c[++t].to=y;
  c[t].next=head[x];
  head[x]=t;
}
inline int sum_H(int x)
{
  int sum=;
  while(x>) sum+=H[x],x-=x&(-x);
  return sum;
}
inline int sum_P(int x)
{
  int sum=;
  while(x>) sum+=P[x],x-=x&(-x);
  return sum;
}
inline void ins_P(int x,int key)
{
  while(x<=n) P[x]+=key,x+=x&(-x);
}
inline void ins_H(int x,int key)
{
  while(x<=n) H[x]+=key,x+=x&(-x);
}
int a[MAXN],pos[MAXN];
int Ans[MAXN][];
int comp(const int x,const int y)
{
  return a[x]>a[y];
}
inline void Init()
{
  scanf("%d",&n);
  for(int i=,x;i<=n;i++)x=read(),add(x,i);
  for(int i=;i<=n;i++)a[i]=read(),pos[i]=i;
  sort(pos+,pos+n+,comp);
  for(int i=;i<=n;i++)a[pos[i]]=i;
}
void dfs(int x)
{
  ins_H(a[x],);
  int y=sum_H(a[x]-);
  Ans[x][]=sum_P(a[x]-);
  ins_P(a[x],);
  for(int i=head[x];i;i=c[i].next)
   dfs(c[i].to);
  Ans[x][]=sum_H(a[x]-)-y;
  ins_P(a[x],-);
  Ans[x][]=a[x]--Ans[x][];
}
inline void work()
{
  dfs();
  for(int i=;i<=n;i++)
   printf("%d %d %d\n",Ans[i][],Ans[i][],Ans[i][]);
}
int main()
{
  Init();
  work();
  return ;
}