【HEOI 2015】兔子与樱花

题目描述

有一棵 n n 个节点组成的有根树，根节点是编号为 0 0 的节点，每个节点有一个权值，这棵树必须保证每个节点上的权值与其子节点的个数之和均不超过 m m 。你可以选择将一个节点删除，一个节点删除后其子节点全部变为其父节点的子节点，且该节点的权值也累加进其父节点中，求最多能删除多少个节点。 n≤2000000 n ≤ 2000000 。

算法分析

对于每个节点，将其删除后对其父节点产生的代价为 val+soncnt−1 v a l + s o n c n t − 1 ，考虑贪心，自底向上计算，对于每个节点将其子节点排序，取代价小的能放则放，时间复杂度 O(nlog2n) O ( n l o g 2 n ) 。

代码实现

#include 
#include 
#include 
const int maxn=2000005;
int n,m,c[maxn];
std::vector<int> g[maxn];
inline bool cmp(int x,int y) {return c[x]int ans=0;
void solve(int x) {
    for(int i=0,end=g[x].size();istd::sort(g[x].begin(),g[x].end(),cmp);
    for(int i=0,end=g[x].size();iint v=g[x][i];
        if(c[x]+c[v]-1<=m) {
            ++ans;
            c[x]+=c[v]-1;
        }
    }
}
int main() {
    scanf("%d%d",&n,&m);
    for(int i=0;iscanf("%d",&c[i]);
    int cnt,x;
    for(int i=0;iscanf("%d",&cnt);c[i]+=cnt;
        while(cnt--) {
            scanf("%d",&x);
            g[i].push_back(x);
        }
    }
    solve(0);
    printf("%d\n",ans);
    return 0;
}