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}