Codeforces Round #358 (Div. 2) C. Alyona and the Tree(如果这棵树合法有一个条件:对于树上任意一个点,它的祖先到它的距离小于等于它自己的值)

传送门:Codeforces Round #358 (Div. 2) C. Alyona and the Tree

---

题意:给你一棵树,树上每个点和每条边的长度都已经知道,
如果这棵树合法有一个条件:对于树上任意一个点,它的祖先到它的距离小于等于它自己的值
问至少删除多少个点使得这棵树合法

---

思路:
其实如果一棵树合法,那么它所删除的点的数目其实是固定的,而如果一个点要删除,
那么他的子树一定全部删除,所以我们可以先进行一次dfs,算出每个点它的子树的点数和
如果一个点删除了,那么它的子树便不用判断,一定删除
接下来就是判断哪些点需要删除,可以维护一个祖先到它的距离的最大值

时间复杂度:O(n)

---

#include<bits/stdc++.h>
using namespace std;
const int maxn=101000;
struct Edge{
    int to,next,cost;
}e[maxn*2];
int ans,sz[maxn],a[maxn],tot,dis[maxn],head[maxn];

void init(){
    tot=0;
    memset(head,-1,sizeof(head));
}

void addedge(int from,int to,int cost){
    e[tot]=(Edge){to,head[from],cost};
    head[from]=tot++;
}

void dfs1(int u,int fa){
    sz[u]=1;
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v==fa)
            continue;
        dfs1(v,u);
        sz[u]+=sz[v];
    }
}

void dfs2(int u,int fa){
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v==fa)
            continue;
        dis[v]=dis[u]+e[i].cost;
        if(dis[v]>a[v]){
            ans+=sz[v];
            continue;
        }
        dis[v]=max(dis[v],0);
        dfs2(v,u);
    }
}

int main(){
    int n,v,cost;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    init();
    for(int i=2;i<=n;i++){
        scanf("%d%d",&v,&cost);
        addedge(i,v,cost);
        addedge(v,i,cost);
    }
    dfs1(1,0);
    dfs2(1,0);
    printf("%d\n",ans);
    return 0;
}