1. 程式人生 > >739B Codeforces Alyona and a tree 樹上差分+二分(倍增)

739B Codeforces Alyona and a tree 樹上差分+二分(倍增)

題目大意是給出一棵有根樹,樹上每個點、每條邊都有一個權值。

現在給出“控制”的定義:對一個點u,設點v在其子樹上,且dis(u,v)av,則稱u控制v。

要求求出每個點控制了多少個點

模擬dfs過程,我們很容易發現dfs到點u時,其祖先節點到根的dis值都已經算出,且是單調遞增,所以我們可以用二分或者倍增,在log的時間複雜度內找到深度最小的滿足dis(u,v)au的點了。找到點v後,v到fa(u)上的每個點的答案都要增加1,顯然用樹上差分來實現,對ans[fa[v]]–,ans[fa[u]]++

#include <cstdio>
#include <cmath>
#include <cstring> #include <string> #include <cstdlib> #include <algorithm> #include <iostream> using namespace std; const int maxn = 200010; const int lg = 20; int n; long long a[maxn]; long long c[maxn]; struct edge { int v,w,next; }e[maxn << 1]; long long dep[maxn]; int
h[maxn],num = 0; int f[maxn][lg+1]; int u,v; long long w; void build(int u,int v,int w) { num++; e[num].v = v; e[num].w = w; e[num].next = h[u]; h[u] = num; } void dfs1(int x,int fa) { int v; f[x][0] = fa; for(int i = 1; i < lg; i++) f[x][i] = f[f[x][i-1]][i-1
]; int u = x; for(int i = lg - 1; i >= 0; i--) while(u > 1 && dep[x] - dep[f[u][i]] <= a[x]) u = f[u][i]; u = max(1,u); c[f[u][0]]--; c[f[x][0]]++; for(int i = h[x]; i; i = e[i].next) { v = e[i].v; if(v != fa) { dep[v] = dep[x] + e[i].w; dfs1(v,x); } } } void dfs2(int x,int f) { int v; for(int i = h[x]; i; i = e[i].next) { v = e[i].v; if(v != f) { dfs2(v,x); c[x] += c[v]; } } } int main() { freopen("1.in","r",stdin); cin >> n; for(int i = 1; i <= n; i++) scanf("%I64d",&a[i]); for(int i = 2; i <= n; i++) { scanf("%d%I64d",&v,&w); build(i,v,w); build(v,i,w); } dfs1(1,0); dfs2(1,0); for(int i = 1; i <= n; i++) printf("%I64d%c",c[i],i == n ? '\n' : ' '); return 0; }