codeforces 381 div 2 D. Alyona and a tree(二分+前缀和)
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}