題目

[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;
}