P3605 [USACO17JAN]Promotion Counting P（树状数组）

解析

做法很多的一道题

sol1

先求出dfs序，离线下来，然后按权值大小的顺序统计答案并插到对应的dfs序中

sol2

离散化后，dfs过程中动态维护树状数组，利用前后差值求出答案

sol3

树上dsu
就比较无脑了，暴力维护即可
但是代码比前两个难写

sol4

权值线段树合并
这个做法在本题中确实是一点优势也没有
兼具了常数大和难编写的双重优势（bushi

总的来说，前两种做法比较推荐

代码

使用的是soltion1
别问我为什么当时还敲了个树剖和线段树上去

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=3e5+100;
int n,m;
struct node{
	int to,nxt;
}p[N<<1];
int fi[N],cnt=-1;
void addline(int x,int y){
	p[++cnt]=(node){y,fi[x]};
	fi[x]=cnt;
}
int dfn[N],fa[N],top[N],pos[N],tim,dep[N],hson[N],siz[N];
void dfs1(int x,int f){
	fa[x]=f;//printf("x=%d f=%d\n",x,f);
	siz[x]=1;
	for(int i=fi[x];~i;i=p[i].nxt){
		int to=p[i].to;
		if(to==f) continue;
		dep[to]=dep[x]+1;
		dfs1(to,x);
		siz[x]+=siz[to];
		if(!hson[x]||siz[hson[x]]<siz[to]) hson[x]=to;
	}
	return;
}
void dfs2(int x,int tp){
	dfn[++tim]=x;pos[x]=tim;top[x]=tp;
	if(hson[x]) dfs2(hson[x],tp);
	for(int i=fi[x];~i;i=p[i].nxt){
		int to=p[i].to;//printf("x=%d to=%d\n",x,to);
		if(to==fa[x]||to==hson[x]) continue;
		dfs2(to,to);
	}
	return;
}
#define mid ((l+r)>>1)
#define ls k<<1
#define rs k<<1|1
ll sum[N<<2],add[N<<2];
void pushup(int k){
	sum[k]=sum[ls]+sum[rs];
}
void Add(int k,int l,int r,ll v){
	add[k]+=v;
	sum[k]+=1ll*(r-l+1)*v;
	return; 
}
void pushdown(int k,int l,int r){
	ll o=add[k];add[k]=0;
	if(!o) return;
	Add(ls,l,mid,o);
	Add(rs,mid+1,r,o);
	return; 
}
void change(int k,int l,int r,int x,int v){
	if(l==r){
		sum[k]+=v;return;
	}
	pushdown(k,l,r);
	if(x<=mid) change(ls,l,mid,x,v);
	else change(rs,mid+1,r,x,v);
	pushup(k);return;
}
void longchange(int k,int l,int r,int x,int y,int v){
	if(x<=l&&r<=y){
		Add(k,l,r,v);return;
	}
	pushdown(k,l,r);
	if(x<=mid) longchange(ls,l,mid,x,y,v);
	if(y>mid) longchange(rs,mid+1,r,x,y,v);
	pushup(k);return;
}
ll ask(int k,int l,int r,int x,int y){
	if(x<=l&&r<=y){
		//printf("k=%d l=%d r=%d x=%d y=%d res=%lld\n",k,l,r,x,y,sum[k]);
		return sum[k];
	}
	pushdown(k,l,r);
	ll res=0;
	if(x<=mid) res+=ask(ls,l,mid,x,y);
	if(y>mid) res+=ask(rs,mid+1,r,x,y);
	//printf("k=%d l=%d r=%d x=%d y=%d res=%lld\n",k,l,r,x,y,res);
	return res;
}
ll sumquery(int x,int y){
	ll res=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		res+=ask(1,1,n,pos[top[x]],pos[x]);
		x=fa[top[x]];
	}
	if(dep[x]<dep[y]) swap(x,y);
	res+=ask(1,1,n,pos[y],pos[x]);
	return res;
}
struct P{
	int id,v;
	bool operator < (const P y)const{return v>y.v;}
}a[N];
int ans[N];
int main(){
	memset(fi,-1,sizeof(fi));
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i].v);a[i].id=i;
	}
	int x,y;
	for(int i=2;i<=n;i++){
		scanf("%d",&x);
		addline(x,i);addline(i,x);
	}
	dfs1(1,0);
	dfs2(1,1);
	//build(1,1,n);
	//for(int i=1;i<=n;i++) printf("i=%d dep=%d fa=%d siz=%d top=%d pos=%d hson=%d\n",i,dep[i],fa[i],siz[i],top[i],pos[i],hson[i]);
	//scanf("%d",&m);
	//int op;
	sort(a+1,a+1+n);
	for(int i=1;i<=n;i++){
		int now=a[i].id;
		ans[now]=ask(1,1,n,pos[now],pos[now]+siz[now]-1);
		change(1,1,n,pos[now],1);
	}
	for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
}
/*

*/