P3605 [USACO17JAN]Promotion Counting晋升者计数

思路

线段树合并的板子。。

和子节点合并之后在值域线段树上查询即可

代码

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN = 1000100;
int n,Nodecnt,root[MAXN],u[MAXN<<1],v[MAXN<<1],cnt,fir[MAXN],nxt[MAXN<<1],ans[MAXN],ax[MAXN],w_p[MAXN],nx;
struct Node{
	int lson,rson,sz;
}Seg[MAXN<<2];
int merge(int x,int y,int l,int r){
	if(x*y==0)
		return x+y;
	if(l==r){
		int t=++Nodecnt;
		Seg[t].sz=Seg[x].sz+Seg[y].sz;
		return Nodecnt;
	}
	int g=++Nodecnt;
	Seg[g].sz=Seg[x].sz+Seg[y].sz;
	int mid=(l+r)>>1;
	Seg[g].lson=merge(Seg[x].lson,Seg[y].lson,l,mid);
	Seg[g].rson=merge(Seg[x].rson,Seg[y].rson,mid+1,r);
	return g;
}
int query(int l,int r,int o,int val){
	if(l==r){
		return 0;
	}
	int mid=(l+r)>>1;
	if(val<=mid)
		return query(l,mid,Seg[o].lson,val)+Seg[Seg[o].rson].sz;
	else
		return query(mid+1,r,Seg[o].rson,val);
}
void build(int l,int r,int &o,int val){
	if(!o)
		o=++Nodecnt;
	Seg[o].sz++;
	if(l==r)
		return;
	int mid=(l+r)>>1;
	if(val<=mid)
		build(l,mid,Seg[o].lson,val);
	else
		build(mid+1,r,Seg[o].rson,val);
}
void addedge(int ui,int vi){
	++cnt;
	u[cnt]=ui;
	v[cnt]=vi;
	nxt[cnt]=fir[ui];
	fir[ui]=cnt;
}
void dfs(int u,int fa){
	for(int i=fir[u];i;i=nxt[i]){
		if(v[i]==fa)
			continue;
		dfs(v[i],u);
		root[u]=merge(root[v[i]],root[u],1,n);
	}
	ans[u]=query(1,n,root[u],w_p[u]);
}
void init(void){
	sort(ax+1,ax+n+1);
	nx=unique(ax+1,ax+n+1)-ax-1;
//	printf("%d\n",nx);
	for(int i=1;i<=n;i++){
		w_p[i]=lower_bound(ax+1,ax+nx+1,w_p[i])-ax;
//		printf("%d!\n",w_p[i]);
	}
	for(int i=1;i<=n;i++){
		build(1,n,root[i],w_p[i]);
	}
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&w_p[i]),ax[i]=w_p[i];
	}
	for(int i=2;i<=n;i++){
		int x;
		scanf("%d",&x);
		addedge(i,x);
		addedge(x,i);
	}
	init();
	dfs(1,0);
	for(int i=1;i<=n;i++)
		printf("%d\n",ans[i]);
	return 0;
}