CF - 739B (Alyona and a tree)

题意：给定一颗树(1为根节点)，每个点有个权值，每条边也有边权，现在给出所有的权值和边权，定义一个点u控制另一个点v的条件：

① u 是 v 的祖先节点；

② u 到 v 上的边权之和 sum 不大于 v 的点权 a[v] ，即 sum<=a[v]；

求每个点控制的节点的数量；

 

分析：先考虑对于每个子节点 v ，找到它最大的祖先 u 满足边权之和 sum<=a[v]，那么 u 到 v 的父节点都可以控制 v，即 u 到 v 的父节点之间的所有点的控制数都+1，到这里是不是可以想到是树上差分，然后找祖先就可以用倍增写了；

 

代码：

#include<vector> 
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
typedef long long ll;
const int N = 2E5+10;

struct node{
	int v,w;
	node(int _v,int _w){
		v=_v,w=_w;
	}
};
vector<node>e[N];

int ans[N],a[N];
int fa[20][N],n;
ll dis[N];

void dfs(int now,int pre,ll sum){
	dis[now]=sum;
	ans[now]=1;
	fa[0][now]=pre;
	for(int i=1;fa[i-1][fa[i-1][now]];i++){
		fa[i][now]=fa[i-1][fa[i-1][now]];
	}
	
	int pos=now;
	for(int i=19;i>=0;i--){
		if(fa[i][pos]&&dis[now]-dis[fa[i][pos]]<=a[now]){
			pos=fa[i][pos];
		}
	}
	pos=fa[0][pos];
	ans[pos]--; 
	for(int i=0;i<e[now].size();i++){
		int to=e[now][i].v,w=e[now][i].w;
		if(to==pre) continue;
		dfs(to,now,sum+w);
		ans[now]+=ans[to];
	}
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	memset(fa,0,sizeof(fa));
	for(int i=2,f,w;i<=n;i++){
		scanf("%d%d",&f,&w);
		e[f].push_back(node(i,w));
	}
	dfs(1,0,0);
	for(int i=1;i<=n;i++){
		printf("%d%c",ans[i]-1,i==n?'\n':' ');
	}
}