BZOJ 4027: [HEOI2015]兔子与樱花 树上dp

点击打开链接

思路：树上dp，每个点的权值为d[i]+G[i].size()，然后选择最小的删除就好

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 2000000+10;

vector<int> G[maxn];
int n,m,d[maxn],ans;

bool cmp(int x,int y){
    return d[x]<d[y];
}

void dfs(int u){
    for(int i=0; i<G[u].size(); i++)
        dfs(G[u][i]);
    d[u] += G[u].size();
    sort(G[u].begin(),G[u].end(),cmp); // 贪心 从重量小的开始拿
    for(int i=0; i<G[u].size(); i++){
        int v = G[u][i];
        if(d[u]+d[v]-1<=m){
            d[u] = d[u]+d[v]-1;
            ans++;
        }
        else
            break;
    }
}

int main(){
    cin>>n>>m;
    for(int i=0; i<n; i++)
        scanf("%d",&d[i]);
    for(int i=0; i<n; i++){
        int num; scanf("%d",&num);
        for(int j=0; j<num; j++){
            int x; scanf("%d",&x);
            G[i].push_back(x);
        }
    }
    ans = 0;
    dfs(0);
    cout << ans << endl;
}