CF 739B - Alyona and a tree

題目大意：一棵樹有n(n≤2∗105) 個點，編號爲1−n ，固定1 號節點爲根。每個點有一個值ai ，樹上的每一條邊的長度爲wi 。對每一個點定義一個支配值ci ，ci 爲i 點爲根的子樹中的滿足到i 的距離不超過aj 的點j的個數。 求每個點的ci 。

題解：比較容易想到，對於每一個點i ，二分它的祖先節點，找到第一個到i 的距離大於ai 的祖先，那麼從祖先到它(不包括該祖先和i 本身)這條鏈上的點的答案都會+1。因爲是靜態更新，所以只要維護好樹上前綴和就可以了。二分祖先節點實際上不需要倍增查找，先祖節點都在dfs的搜索棧中，直接開一個棧，在棧上二分就可以。前綴和需要從底向上累加，其實在退棧的時候順便更新就好了。

//= = 不該用倍增的
#include <bits/stdc++.h>

using namespace std;

const int maxn = 200020;
typedef long long ll;
struct node{
    int v;
    int val;
    node(int v,int val):v(v),val(val){}
    node(){}
    bool operator < (const node & a) const{
        return val>a.val;
    }
};
vector<node> T[maxn];

int pa[maxn][20];
ll dis[maxn];
int dep[maxn];
int a[maxn];
int ans[maxn];
node po[maxn];

void dfs(int u){
    ll d = dis[u];
    for(int i = 0;i < T[u].size();i++){
        int v = T[u][i].v;
        dis[v] = d + T[u][i].val;
        dep[v] = dep[u] + 1;
        dfs(v);
    }
}

int findfa(int u,int d){
    int x = u;
    for(int i = 19;i >=0 && d;i--){
        if((1<<i)<=d){
            x = pa[x][i];
            d -= (1<<i);
        }
    }
    return x;
}

int main(){
    int n;
    scanf("%d",&n);
    for(int i = 1;i <= n;i++){
        scanf("%d",&a[i]);
    }
    int p,w;
    dis[1] = 0;
    dep[1] = 0;
    memset(pa,-1,sizeof(pa));
    memset(ans,0,sizeof(ans));
    pa[1][0] = 0;
    for(int i = 2;i <= n;i++){
        scanf("%d%d",&p,&w);
        T[p].push_back(node(i,w));
        pa[i][0] = p;
    }
    dfs(1);
    po[0].v = 0;
    po[0].val = -1;
    for(int i = 1;i <= n;i++){
        po[i].v = i;
        po[i].val = dep[i];
    }
    for(int j = 1;j < 20;j++){
        for(int i = 1;i <= n;i++){
            if(~pa[i][j-1])
            pa[i][j] = pa[pa[i][j-1]][j-1];
        }
    }

    sort(po,po+1+n);
    for(int i = 2;i <= n;i++){
        int l = 0,r = dep[i],mid;
        while(l < r){
            mid = (l+r+1)>>1;
            if(dis[i]-dis[findfa(i,mid)]>a[i]) r = mid-1;
            else l = mid;
        }
        int t = pa[findfa(i,l)][0];
        ans[pa[i][0]]++;
        ans[t]--;
    }
    for(int i = 0;i <= n;i++){
        int v,p;
        v = po[i].v;
        p = pa[v][0];

        if(p != -1){
            ans[p] += ans[v];
        }
    }
    for(int i = 1;i <= n;i++){
        printf("%d",ans[i]);
        if(i == n) printf("\n");
        else printf(" ");
    }

    return 0;
}