bzoj 4027 兔子与樱花

Posted on | Edited on | Views:

贪心.

在合法的前提下,一定是在每棵子树中也删掉尽可能多的点,这样这个子树的根对它的父亲的贡献最小.

于是对树 $dfs$ ,处理完 $u$ 的所有子节点 $v$ 后,需要考虑删掉哪些 $v$ .

将每个 $v$ 被删掉后对 $u$ 的贡献从小到大排序,能删就删即可.

时间复杂度 $O(n\log n)$ .

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
inline int read()
{
    int out=0,fh=1;
    char jp=getchar();
    while ((jp>'9'||jp<'0')&&jp!='-')
        jp=getchar();
    if (jp=='-')
        fh=-1,jp=getchar();
    while (jp>='0'&&jp<='9')
        out=out*10+jp-'0',jp=getchar();
    return out*fh;
}
const int MAXN=2e6+10;
int ecnt=0,nx[MAXN],to[MAXN],head[MAXN];
void addedge(int u,int v)
{
    ++ecnt;
    to[ecnt]=v;
    nx[ecnt]=head[u];
    head[u]=ecnt;
}
int n,m,c[MAXN],ans=0,tmp[MAXN],tot;
int sonsiz[MAXN];
bool cmp(int x,int y)
{
    return c[x]+sonsiz[x]<c[y]+sonsiz[y];
}
void dfs(int u)
{
    for(int i=head[u];i;i=nx[i])
    {
        int v=to[i];
        dfs(v);
    }
    tot=0;
    for(int i=head[u];i;i=nx[i])
    {
        int v=to[i];
        tmp[++tot]=v;
    }
    sort(tmp+1,tmp+1+tot,cmp);
    for(int i=1;i<=tot;++i)
    {
        int v=tmp[i];
        if(c[u]+sonsiz[u]+c[v]+sonsiz[v]-1<=m)
        {
            c[u]+=c[v];
            sonsiz[u]+=sonsiz[v]-1;
            ++ans;
        }
    }
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;++i)
        c[i]=read();
    for(int i=1;i<=n;++i)
    {
        sonsiz[i]=read();
        for(int k=1;k<=sonsiz[i];++k)
        {
            int j=read()+1;
            addedge(i,j);
        }
    }
    dfs(1);
    cout<<ans<<endl;
    return 0;
}