[HEOI 2015] 兔子与樱花（贪心） | 错题本

题目

[HEOI 2015] 兔子与樱花

分析

从叶子往根做，考虑选取 $u$ 子节点 $v$ 中 $w(v)=son(v)+c_v$ 小的优先删除，因为假如此时不删 $w(v)$ 最小而删 $w(v)$ 更大的 $v$，必然会使 $u$ 更容达到到 $m$ 的限制，于是删除的节点数不会多于删最小的。

代码

#include <bits/stdc++.h>

int Read() {
	int x = 0; char c = getchar();
	while (c < '0' || c > '9')
		c = getchar();
	while (c >= '0' && c <= '9')
		x = x * 10 + (c ^ 48), c = getchar();
	return x;
}

const int MAXN = 2000000;

int N, M, Ans;
int C[MAXN + 5], K[MAXN + 5];
std::vector<int> G[MAXN + 5];

inline bool Comp(const int &i, const int &j) {
	return K[i] + C[i] < K[j] + C[j];
}

void Dfs(const int &u) {
	for (int v: G[u])
		Dfs(v);
	std::sort(G[u].begin(), G[u].end(), Comp);
	for (int v: G[u]) {
		if (K[u] + K[v] - 1 + C[u] + C[v] <= M)
			K[u] += K[v] - 1, C[u] += C[v], Ans++;
		else break;
	}
}

int main() {
	N = Read(), M = Read();
	for (int i = 0; i < N; i++)
		C[i] = Read();
	for (int i = 0; i < N; i++) {
		K[i] = Read();
		for (int j = 1; j <= K[i]; j++)
			G[i].push_back(Read());
	}
	Dfs(0);
	printf("%d", Ans);
	return 0;
}