HeRaNO's NOIP CSP Round Day 2 T3 ginkgo

 

 

 

 

 睡醒后我第一眼：这不主席树裸题吗？

先统计dfs序，把树上问题转化为区间问题

区间大于等于某个数的个数。。。主席树模板？

#include<bits/stdc++.h>
#define re register
#define inc(i,j,k) for(re int i=j;i<=k;++i)
#define dec(i,j,k) for(re int i=j;i>=k;--i)
#define ra(i,u) for(re int i=head[u];i;i=e[i].nxt)
using namespace std;
const int maxn=200010;
inline int read()
{
	re int x=0,f=1; char ch=getchar();
	while(ch<'0'||ch>'9') {if(ch=='-') f=-1; ch=getchar();}
	while(ch>='0'&&ch<='9') {x=x*10+(ch^48); ch=getchar();}
	return x*f;
}
int L[maxn*20],R[maxn*20],sum[maxn*20];
int T[maxn],b[maxn],a[maxn],size[maxn],dfn[maxn<<1],head[maxn];
int n,m,tot,num,tim,cnt;
struct node
{
	int to,nxt;
}e[maxn<<1];
inline void add(int u,int v)
{
	e[++cnt].to=v;
	e[cnt].nxt=head[u];
	head[u]=cnt;
}
inline int update(int pre,int l,int r,int v)
{
	re int p=++tot;
	L[p]=L[pre]; R[p]=R[pre];
	sum[p]=sum[pre]+1;
	if(l==r) return p;
	re int mid=(l+r)>>1;
	if(v<=mid) L[p]=update(L[pre],l,mid,v);
	else R[p]=update(R[pre],mid+1,r,v);
	return p;
}
inline int query(int p,int pre,int l,int r,int x,int y)
{
	if(x<=l && r<=y) return sum[p]-sum[pre];
	re int mid=(l+r)>>1;
	re int res=0;
	if(x<=mid) res+=query(L[p],L[pre],l,mid,x,y);
	if(y>mid) res+=query(R[p],R[pre],mid+1,r,x,y);
	return res;
}
inline void dfs(int u,int fa)
{
	size[u]=1;
	dfn[u]=++tim;
	T[tim]=update(T[tim-1],0,num+1,a[u]);
	ra(i,u)
	{
		re int v=e[i].to;
		if(v==fa) continue;
		dfs(v,u);
		size[u]+=size[v];
	}
}
int main()
{
//	freopen("123.in","r",stdin);
	re int xx,yy;
	n=read();
	inc(i,2,n)
	{
		xx=read();
		add(xx,i);
	}
	inc(i,1,n) a[i]=read(),b[i]=a[i];
	sort(b+1,b+n+1);
	num=unique(b+1,b+n+1)-b-1;
	inc(i,1,n)
	{
		a[i]=lower_bound(b+1,b+num+1,a[i])-b;
	}
	dfs(1,0);
	inc(i,1,n)
	{
		xx=T[dfn[i]]; yy=T[dfn[i]+size[i]-1];
//		cout<<xx<<" "<<yy<<endl;
		printf("%d %d %d\n",query(yy,xx,0,num+1,0,a[i]-1),query(yy,xx,0,num+1,a[i],a[i]),query(yy,xx,0,num+1,a[i]+1,num+1));
	}
}

 

 小（大）插曲

考完试。

老师:诶要不hxx你来讲讲这套题吧，你为啥答得这么快啊

hxx（这是位初四的可爱弟弟）：因为这套题太水了，我本来看我迟到了以为答不完了，结果还好在一个多小时写完了。。。

完。

(Orz hxx JULAO