BZOJ_4756_[Usaco2017 Jan]Promotion Counting_树状数组

BZOJ_4756_[Usaco2017 Jan]Promotion Counting_树状数组

Description

n只奶牛构成了一个树形的公司，每个奶牛有一个能力值pi，1号奶牛为树根。
问对于每个奶牛来说，它的子树中有几个能力值比它大的。

Input

n，表示有几只奶牛 n<=100000
接下来n行为1-n号奶牛的能力值pi
接下来n-1行为2-n号奶牛的经理（树中的父亲）

Output

共n行，每行输出奶牛i的下属中有几个能力值比i大

Sample Input

5
804289384
846930887
681692778
714636916
957747794
1
1
2
3

Sample Output

2
0
1
0
0   这道题的思路比较巧妙。 我们对整棵树进行DFS时每个点有两次处理的机会。 一次是刚刚遍历到，一次是子树的回溯。 分别求出两次能力值大于这个点的点的个数，用第二次的减去第一次的就是这个点的答案。 然后离散用树状数组处理一下。   代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 100050
struct A {
int num,id,v;
}a
;
bool cmp1(const A &x,const A &y){return x.num>y.num; }
bool cmp2(const A &x,const A &y){return x.id<y.id; }
int n,head
,to[N<<1],nxt[N<<1],cnt,ans
,c
;
inline void add(int u,int v) {
to[++cnt]=v;nxt[cnt]=head[u];head[u]=cnt;
}
void fix(int x,int v) {
for(;x<=n;x+=x&-x) c[x]+=v;
}
int inq(int x) {
int re=0;
for(;x;x-=x&-x) re+=c[x];
return re;
}
void dfs(int x,int y) {
int i;
fix(a[x].v,1);
int tmp=inq(a[x].v-1);
for(i=head[x];i;i=nxt[i]) {
if(to[i]!=y) {
dfs(to[i],x);
}
}
ans[x]=inq(a[x].v-1)-tmp;
}
int main() {
//freopen("tt.in","r",stdin);
scanf("%d",&n);
int i,x,j;
for(i=1;i<=n;i++) scanf("%d",&a[i].num),a[i].id=i;
sort(a+1,a+n+1,cmp1); a[0].num=-245345;
for(i=1,j=0;i<=n;i++) { if(a[i].num!=a[i-1].num) j++; a[i].v=j; }
sort(a+1,a+n+1,cmp2);
for(i=2;i<=n;i++) {
scanf("%d",&x);
add(i,x); add(x,i);
}
dfs(1,0);
for(i=1;i<=n;i++) printf("%d\n",ans[i]);
}