BZOJ4027: [HEOI2015]兔子与樱花

BZOJ4027

一开始想的都是二维的$dp$。。数据这么大怎么转移啊$QAQ$。
后来发现原来贪心就行辣。。。
直接记$ans_i=son_i+c_i$，那么将这个点删除对应着其父节点$+=ans_i-1$
那么对于某个点$x$，将儿子的$ans$值从小到大排徐，然后依次删直到不能删为止。
然后删下面的比删上面的优。
因为删某个点$x$的儿子节点，有影响的只是当前点$x$，最多只会造成当前点无法删掉。但是儿子节点中至少删一个，所以不亏。为什么对其他点没有影响呢？
考虑$x$的兄弟节点$xx$，因为$ans_x$变大导致$x$无法删除，只会使兄弟节点被删除的更多。
考虑$x$的父节点$fa_x$，如果$ans_x$的增大仅仅造成了少删$x$一个，那么$ans_{fa_x}$相较会变小，也就是不会加上$ans_x-1$，所以被删除的机会会更大。
另外如果$ans_x$增大使得兄弟节点中有至少一个被多删除掉。就算使得$ans_{fa_x}$相较变大而导致无法删掉，也是不亏的。
所以就可以开开心心的贪心辣！

【代码】

#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
#define N 2000005
using namespace std;
typedef long long ll;

ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,ans;
int c[N],sz[N];
vector<int>g[N]; 

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

void Dfs(int x)
{
    for(int i=0;i<g[x].size();i++) Dfs(g[x][i]);
    sort(g[x].begin(),g[x].end(),cmp);
    for(int i=0;i<g[x].size();i++) {
        int v=c[g[x][i]];
        if(c[x]+v-1<=m) c[x]+=v-1,ans++;
        else break;
    }
}

int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++) c[i]=read();
    for(int i=1;i<=n;i++)
    {
        int num=read();c[i]+=num;
        for(int j=1;j<=num;j++) g[i].push_back(read()+1);
    }
    Dfs(1);
    printf("%d\n",ans);
    return 0;
}