洛谷 P3605 [USACO17JAN]Promotion Counting P 题解--zhengjun

思路

直接 dfs 一遍。

首先，一个子树内的点都是一遍 dfs 到的，那么直接差分一下，到达 $u$ 的时候记录当前的 $>p_u$ 的个数，然后 dfs 完 $u$ 的子树的时候，记录 $>p_u$ 的个数，相减即为 $u$ 的答案。

代码

#include<bits/stdc++.h>
using namespace std;typedef long long ll;const int N=1e5+10;
int n,a[N],b[N],c[N],ans[N];vector<int>e[N];
void add(int x,int y){for(;x;x-=(x&-x))c[x]+=y;}
int get(int x){int y=0;for(;x<=b[0];x+=(x&-x))y+=c[x];return y;}
void dfs(int u){ans[u]-=get(a[u]);add(a[u]-1,1);for(int v:e[u])dfs(v);ans[u]+=get(a[u]);}
int main(){
	scanf("%d",&n);for(int i=1;i<=n;i++)scanf("%d",&a[i]),b[i]=a[i];sort(b+1,b+1+n);b[0]=unique(b+1,b+1+n)-b-1;
	for(int i=1;i<=n;i++)a[i]=lower_bound(b+1,b+1+b[0],a[i])-b;
	for(int i=2,x;i<=n;i++)scanf("%d",&x),e[x].push_back(i);
	dfs(1);for(int i=1;i<=n;i++)printf("%d\n",ans[i]);return 0;
}