题目描述
有一棵 个节点组成的有根树,根节点是编号为 的节点,每个节点有一个权值,这棵树必须保证每个节点上的权值与其子节点的个数之和均不超过 。你可以选择将一个节点删除,一个节点删除后其子节点全部变为其父节点的子节点,且该节点的权值也累加进其父节点中,求最多能删除多少个节点。 。
算法分析
对于每个节点,将其删除后对其父节点产生的代价为 ,考虑贪心,自底向上计算,对于每个节点将其子节点排序,取代价小的能放则放,时间复杂度 。
代码实现
#include <cstdio>
#include <vector>
#include <algorithm>
const int maxn=2000005;
int n,m,c[maxn];
std::vector<int> g[maxn];
inline bool cmp(int x,int y) {return c[x]<c[y];}
int ans=0;
void solve(int x) {
for(int i=0,end=g[x].size();i<end;++i) solve(g[x][i]);
std::sort(g[x].begin(),g[x].end(),cmp);
for(int i=0,end=g[x].size();i<end;++i) {
int v=g[x][i];
if(c[x]+c[v]-1<=m) {
++ans;
c[x]+=c[v]-1;
}
}
}
int main() {
scanf("%d%d",&n,&m);
for(int i=0;i<n;++i) scanf("%d",&c[i]);
int cnt,x;
for(int i=0;i<n;++i) {
scanf("%d",&cnt);c[i]+=cnt;
while(cnt--) {
scanf("%d",&x);
g[i].push_back(x);
}
}
solve(0);
printf("%d\n",ans);
return 0;
}