记录编号	314381	评测结果	AAAAAAAAAA
题目名称	树的统计	最终得分	100
用户昵称	AntiLeaf	是否通过	通过
代码语言	C++	运行时间	0.159 s
提交时间	2016-10-03 08:07:17	内存使用	2.58 MiB

#include<cstdio>
#include<cstring>
#include<algorithm>
#define lowbit(x) ((x)&(-(x)))
using namespace std;
namespace mine{
	template<class T>inline void readint(T &__x){
		static int __c;
		static bool __neg;
		__x=0;
		__neg=false;
		do __c=getchar();while(__c==' '||__c=='\n'||__c=='\r'||__c=='\t');
		if(__c=='-'){
			__neg=true;
			__c=getchar();
		}
		for(;__c>='0'&&__c<='9';__c=getchar())__x=__x*10+(__c^48);
		if(__neg)__x=-__x;
	}
	template<class T>inline void putint(T __x){
		static int __a[40],__i,__j;
		static bool __neg;
		__neg=__x<0;
		if(__neg)__x=-__x;
		__i=0;
		do{
			__a[__i++]=__x%(T)10^(T)48;
			__x/=10;
		}while(__x);
		if(__neg)putchar('-');
		for(__j=__i-1;__j^-1;__j--)putchar(__a[__j]);
	}
}
using namespace mine;
const int maxn=100010;
struct edge{
	int to,prev;
}e[maxn];
void insert(int,int);
void dfs(int);
void add(int);
int query(int);
int last[maxn]={0},prt[maxn],c[maxn];
int n,len=0,ans[maxn],rt;
int main(){
#define MINE
#ifdef MINE
	freopen("counttree.in","r",stdin);
	freopen("counttree.out","w",stdout);
#endif
	readint(n);
	for(int i=1;i<=n;i++){
		readint(prt[i]);
		if(prt[i])insert(prt[i],i);
		else rt=i;
	}
	dfs(rt);
	for(int i=1;i<=n;i++){
		putint(ans[i]);
		putchar(' ');
	}
#ifndef MINE
	printf("\n--------------------DONE--------------------\n");
	for(;;);
#endif
	return 0;
}
inline void insert(int x,int y){
	e[++len].to=y;
	e[len].prev=last[x];
	last[x]=len;
}
inline void dfs(int x){
	ans[x]=-query(x-1);
	add(x);
	for(int i=last[x];i;i=e[i].prev){
		if(e[i].to==prt[x])continue;
		dfs(e[i].to);
	}
	ans[x]+=query(x-1);
}
inline void add(int x){
	while(x<=n){
		c[x]++;
		x+=lowbit(x);
	}
}
inline int query(int x){
	int ans=0;
	while(x){
		ans+=c[x];
		x-=lowbit(x);
	}
	return ans;
}