【BZOJ4027】兔子与樱花（贪心）

【BZOJ4027】兔子与樱花（贪心）

题面

BZOJ
洛谷

题解

很直观的一个感受就是对于每个节点，
考虑它的所有儿子，如果能删就删。
那么我们把所有儿子按照给删去后给父亲\(c[i]\)的贡献从小往大排序，
能删就删即可。

#include<cstdio>
#include<algorithm>
using namespace std;
#define ll long long
#define MAX 2000020
inline int read()
{
    int x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
int n,m,f[MAX],c[MAX],s[MAX];
struct Line{int v,next;}e[MAX];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
void dfs(int u)
{
    for(int i=h[u];i;i=e[i].next)dfs(e[i].v),f[u]+=f[e[i].v];
    int son=0;
    for(int i=h[u];i;i=e[i].next)s[++son]=c[e[i].v]-1;
    sort(&s[1],&s[son+1]);
    for(int i=1;i<=son;++i)
        if(c[u]+s[i]<=m)c[u]+=s[i],++f[u];
        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 k=read();c[i]+=k;
        while(k--)Add(i,read()+1);
    }
    dfs(1);printf("%d\n",f[1]);
    return 0;
}