CF739B Alyona and a tree

我们可以发现，每一个节点，都会向上 对连续的一个节点段中的每个节点产生1的贡献。所以，我们可以对于每个节点先倍增处理出dis<=a[u]的最上面的节点，之后对于这一连续的节点段，差分累加答案。

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+5;
int n,u,w;
int a[N],d[N],dis[N][21],p[N][21],f[N];
int cnt,head[N];
struct edge{int next,to,w;}e[N];

inline void add(int u,int v,int w)
{
	cnt++;
	e[cnt].next=head[u];
	e[cnt].to=v;
	e[cnt].w=w;
	head[u]=cnt;
}

void dfs(int u)
{
	for (register int i=1; (1<<i)<=d[u]; ++i) p[u][i]=p[p[u][i-1]][i-1];
	for (register int i=1; (1<<i)<=d[u]; ++i) dis[u][i]=dis[p[u][i-1]][i-1]+dis[u][i-1]; 
	for (register int i=head[u]; i; i=e[i].next)
	{
		d[e[i].to]=d[u]+1;
		dis[e[i].to][0]=e[i].w;
		p[e[i].to][0]=u;
		dfs(e[i].to);
	}
}

void dfs2(int u)
{
	for (register int i=head[u]; i; i=e[i].next)
	{
		dfs2(e[i].to);
		f[u]+=f[e[i].to];
	}
}

signed main(){
	scanf("%lld",&n);
	for (register int i=1; i<=n; ++i) scanf("%lld",&a[i]);
	for (register int i=2; i<=n; ++i) scanf("%lld%lld",&u,&w),add(u,i,w);	
	dfs(1);
	//for (register int i=1; i<=n; ++i) printf("%lld ",dis[i][0]); puts(""); 
	for (register int i=1; i<=n; ++i)
	{
		bool jay=false;
		int now=i;
		int last=a[i];
		for (register int j=20; j>=0; --j) 
		if (dis[now][j]<=last) 
		{
			if (!p[now][j]) continue; 
			jay=true,last-=dis[now][j],now=p[now][j];
		}
		if (jay) 
		{
			f[p[i][0]]++,f[p[now][0]]--;
		}
	}
	dfs2(1);
	for (register int i=1; i<=n; ++i) printf("%lld ",f[i]);
return 0;
}