Note: This article is available in Chinese only. 本文暂无英文版本。
View original
d：题意：一棵树，给出边权和点权，定义点v控制点u，当且仅当u是v的子树中的点，并且dis(u,v)<=a[u]，其中dis(u,v)为点u到点v路径上的边权和，a[u]为点u的点权，现在问对于每个节点v，其能控制的点有多少个。
思路：先写了个rmq+dfs的lca。。。那么任意两个点的距离都可以O(1)得到了。然后不会了233333.
upd：和lca没有什么关系，因为一个点能控制另一个点这两个点一定在一条通向根的链上，因此距离直接减一下就好了。
机智的做法：dfs的时候维护一个栈，对于栈中序列，后面一半是对当前点有贡献的。问题时求对于每个v统计其能控制多少个u，现在我们固定u，考虑能控制他的v。这些v在树上的形态时一条链 ，借助第二类前缀和的思想，对于u标记+1，对于u往上的离根最近的且能统治u的v上面的一个标记-1，然后dfs后序遍历（也就是链的起点时距离根远的那一边），距离处理的时候，只需要在递归之后更新ans就好了。
栈里面维护，到哪个节点，从根下来，边权和最大，找边权和>=当前边权和-a[u]的地方。
启示：由于两个存在统治关系的点在一条链上，边权都为正，边权和具有单调性，单调的东西，容易想到二分处理。
代码实现
1/* ***********************************************
2Author :111qqz
3Created Time :2016年11月24日 星期四 09时17分48秒
4File Name :code/cf/#381/D.cpp
5************************************************ */
6#include <cstdio>
7#include <cstring>
8#include <iostream>
9#include <algorithm>
10#include <vector>
11#include <queue>
12#include <set>
13#include <map>
14#include <string>
15#include <cmath>
16#include <cstdlib>
17#include <ctime>
18#define fst first
19#define sec second
20#define lson l,m,rt<<1
21#define rson m+1,r,rt<<1|1
22#define ms(a,x) memset(a,x,sizeof(a))
23typedef long long LL;
24#define pi pair < int ,int >
25#define MP make_pair
26using namespace std;
27const double eps = 1E-8;
28const int dx4[4]={1,0,0,-1};
29const int dy4[4]={0,-1,1,0};
30const int inf = 0x3f3f3f3f;
31const int N=2E5+7;
32int n;
33int a[N];
34LL dis[N];
35vector< pair<int,LL> >edge[N];
36vector< pair<LL,int> >path;
37int ans[N];
38void dfs( int u,int pre)
39{
40 ans[u]++;
41 int idx = lower_bound(path.begin(),path.end(),make_pair(dis[u]-a[u],-1))-path.begin()-1;
42 if (idx>=0) ans[path[idx].sec]--;
43 path.push_back(make_pair(dis[u],u));
44 for ( auto x : edge[u])
45 {
46 int v = x.first;
47 if (v==pre) continue;
48 dis[v] = dis[u] + x.sec;
49 dfs(v,u);
50 ans[u]+=ans[v];//后序遍历...链的起点原理根，终点靠近根
51 }
52 path.pop_back();
53}
54int main()
55{
56 #ifndef ONLINE_JUDGE
57 freopen("code/in.txt","r",stdin);
58 #endif
59 cin>>n;
60 ms(dis,0);
61 for ( int i = 1 ; i <= n ; i++) scanf("%d",&a[i]);
62 for ( int i =2 ; i <= n ; i++)
63 {
64 int u,v;
65 LL w;
66 u = i ;
67 scanf("%d%lld",&v,&w);
68 edge[u].push_back(make_pair(v,w));
69 edge[v].push_back(make_pair(u,w));
70 }
71 dfs(1,0);
72 for ( int i = 1 ; i <= n ; i++) printf("%d ",ans[i]-1);
73 #ifndef ONLINE_JUDGE
74 fclose(stdin);
75 #endif
76 return 0;
77}