好吧这题我也没想出来。我果然是个zz。。
很妙的贪心,(其实我也知道贪心,但是姿势不对),我们观察数据n<=2000000,如果树形dp要o(n)很难yy到简便的状态和转移方程,很自然的想到贪心。我们取一个子树分析,发现儿子删除对父亲的贡献为儿子的height+1,那么贡献只与儿子的weight有关,那么就贪心,对每一个儿子进行排序不停取最小的直到无法删除,更新总共删除节点数。
TIP:遇到树一类的思博题,我们要对子树分析,观察数据范围以及询问的ans的特点(要想到树上dp或贪心或用数据结构·维护dfs序)。贪心也要尝试猜一些结论。
代码:
#include<iostream>
#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
const int Maxn=2000005;
int w[Maxn],ns[Maxn];
int sum,n,m;
vector<int>son[Maxn];
int cmp(int a,int b)
{
return w[a]<w[b];
}
int dfs(int x)
{
for(int i=0;i<ns[x];i++)
dfs(son[x][i]);
sort(son[x].begin(),son[x].end(),cmp);
w[x]+=ns[x];
for(int i=0;i<ns[x];i++)
{
if(w[x]+w[son[x][i]]-1<=m)
{
w[x]+=(w[son[x][i]]-1);
sum++;
}
else
break;
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&w[i]);
for(int i=1;i<=n;i++)
{
int x,s;
scanf("%d",&x);
ns[i]=x;
for(int j=1;j<=x;j++)
{
scanf("%d",&s);
son[i].push_back(s+1);
}
}
dfs(1);
printf("%d",sum);
return 0;
}