1. 程式人生 > 其它 >[題解][UOJ-139]被刪除的黑白樹

[題解][UOJ-139]被刪除的黑白樹

\(\Longrightarrow\)原題連結

題意:給定一顆有\(n\)個結點的樹,初始時所有節點都是白色的,現在要將一些節點染成黑色,要求所有葉子節點到根節點的路徑中黑色節點的個數相同。問最多可以將多少個節點染成黑色。

\((u,v)\)表示\(u\)\(v\)的一條路徑,用\(w(u,v)\)表示路徑\((u,v)\)上的黑色節點個數。貪心地想,可以發現對於所有葉子節點\(u\),滿足:

\[lim = w(root, u)=\min_{v \in \mathrm{leaves}} \{dep(v)\} \]

用人話來講,就是每個葉子節點到根的路徑上的黑色節點個數由深度最小的葉子節點確定。

接著可以發現,黑色節點儘量往深度較大的位置放會比較好,也就是說,當我們dfs到了節點\(u\),現在我們在\((u,root)\)上已經放了\(a\)個黑節點,但是在\(u\)點還沒有確定放不放,那麼只有當\(u\)距離在\(T(u)\)內部的最近的葉子節點的距離等於\(lim-a\)才放。

#include <bits/stdc++.h>

using namespace std;

const int maxn = 1e5 + 5;
struct Edge { ... } E[maxn << 1];
int hd[maxn], tote;

void addedge(int u, int v) { ... }

int n, ans, lf[maxn], lf_sz, dep[maxn], f[maxn];

void init(int u, int fa) {
	dep[u] = dep[fa] + 1, f[u] = n + 1;
	bool is_leaf = true;
	for (int i = hd[u]; i; i = E[i].nex) {
		int v = E[i].v;
		if (v == fa) continue;
		init(v, u), is_leaf = false;
		f[u] = min(f[u], f[v]);
	}
	if (is_leaf) f[u] = dep[u];
}
void calcans(int u, int fa, int cnt) {
	if (f[u] - dep[u] < f[1] - cnt) ans++, cnt++;
	for (int i = hd[u]; i; i = E[i].nex) {
		int v = E[i].v;
		if (v == fa) continue;
		calcans(v, u, cnt);
	}
}

int main() {
	scanf("%d", &n);
	for (int i = 1; i < n; i++) {
		int u, v; scanf("%d%d", &u, &v);
		addedge(u, v);
	}
	init(1, 0);
	calcans(1, 0, 0);
	printf("%d\n", ans);
	return 0;
}