Codeforces Round #381 (Div. 1) B. Alyona and a tree (树上差分）

最新推荐文章于 2024-03-13 22:20:21 发布

为什么他们cf写的这么快

最新推荐文章于 2024-03-13 22:20:21 发布

阅读量89

点赞数

分类专栏：差分

本文链接：https://blog.csdn.net/qq_42479630/article/details/104784437

版权

差分专栏收录该内容

7 篇文章 0 订阅

订阅专栏

题目链接
在这里插入图片描述

思路：如果v收u的控制，那么在u->v的路径中的所有点都可以控制v，，点数可以回溯求，只要在到点u的时候二分一下这条链路径上的第一个大于等于dis【u】-a【u】的点，在那个位置减1，差分一下就好。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+1;
ll w,a[maxn],dis[maxn],ans[maxn];
vector<pair<ll,ll> >g[maxn];
vector<pair<ll,int> >q;
void dfs(int u,int fa)
{
	ans[u]=1;
	int pos=lower_bound(q.begin(),q.end(),make_pair(dis[u]-a[u],0))-q.begin();
	pos--;
	if(pos>=0) ans[q[pos].second]--;
	q.push_back(make_pair(dis[u],u));
	for(auto to:g[u])
	{
		if(to.first==fa) continue;
		dis[to.first]=dis[u]+to.second;
		dfs(to.first,u);
		ans[u]+=ans[to.first];
	}
	q.pop_back();
}
int main()
{
	int n,u;
	scanf("%d",&n);
	for(int i=1;i<=n;++i) scanf("%lld",&a[i]);
	for(int i=1;i<n;++i)
	{
		scanf("%d%lld",&u,&w);
		g[i+1].push_back({u,w});
		g[u].push_back({i+1,w});
	}
	dfs(1,-1);
	for(int i=1;i<=n;++i) printf("%d ",ans[i]-1);
}