Luogu

分析

容易想到,对于某一个点,我们一定会先删掉 $c_v+k_v$ 小的儿子。

然后一个奇妙的结论是,我们直接从下往上、能删就删即可。因为删掉一个点对答案的贡献为 $1$,而对父节点的影响至多为 $1$,因此这样子是最优的。

代码

// ===================================
//   author: M_sea
//   website: http://m-sea-blog.com/
// ===================================
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <vector>
#include <cmath>
#define re register
using namespace std;

inline int read() {
    int X=0,w=1; char c=getchar();
    while (c<'0'||c>'9') { if (c=='-') w=-1; c=getchar(); }
    while (c>='0'&&c<='9') X=X*10+c-'0',c=getchar();
    return X*w;
}

const int N=2000000+10;

int n,lim,ans=0;
int c[N],k[N];

struct edge { int v,nxt; } e[N];
int head[N];
inline void addEdge(int u,int v) {
    static int cnt=0;
    e[++cnt]=(edge){v,head[u]},head[u]=cnt;
}

vector<int> vec;
inline int cmp(int x,int y) { return c[x]+k[x]<c[y]+k[y]; }
inline void dfs(int u) {
    for (re int i=head[u];i;i=e[i].nxt) dfs(e[i].v);
    vec.clear();
    for (re int i=head[u];i;i=e[i].nxt) vec.push_back(e[i].v);
    sort(vec.begin(),vec.end(),cmp);
    for (re int v:vec)
        if (c[u]+k[u]+c[v]+k[v]-1<=lim) ++ans,c[u]+=c[v],k[u]+=k[v]-1;
}

int main() {
    n=read(),lim=read();
    for (re int i=1;i<=n;++i) c[i]=read();
    for (re int i=1;i<=n;++i) { k[i]=read();
        for (re int j=1;j<=k[i];++j) addEdge(i,read()+1);
    }
    dfs(1); printf("%d\n",ans);
    return 0;
}
最后修改:2019 年 10 月 25 日 10 : 53 AM