hdu5905 Black White Tree

题意：一棵树已经进行了黑白染色，现在你要找是否存在一个联通子树包含a个白色节点和b个黑色节点，如果存在就把(a+1)*(b+1)累加到答案里

题解：正如官方题解所述，对于一个大小为T的联通子树，最多黑色节点为i，最少为j，那么一定存在一个黑色节点数为k(i<=k<=j)的联通子树，所以我们用常规树DP求最大最小黑点树就行了。关于这个树DP，一开始我以为复杂度会很高，因为每次两颗子树合并的时候复杂度是两颗子树大小的乘积，速度肯定比复杂度为两颗子树大小相加的时候要慢的多。但是参考别人ac代码后发现都是这样DP过去的，所以我重新仔细的算了一下后发现实际复杂度如下：

	全是叶子节点的树	完全二叉树	退化成链的树	平均情况
两颗子树大小相加	O(n)	O(nlogn)	O(n^2)	O(nlogn)
两颗子树大小相乘	O(n^2)	O(n^2)	O(n^2)	O(n^2)

可以看到，相乘条件下确实比相加的时候慢，但是最坏情况下时间复杂度都是n^2的，所以这题实际复杂度是n^2的。并且在每次合并的时候先合并小的能优化不少时间。

代码：

#include <stdio.h>
#include <algorithm>
#include <set>
#include <string.h>
#include <math.h>
#include <iostream>
#include <vector>
#include <map>
#include <queue>
#include <bitset>
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;
const int inf = 0x3f3f3f3f;
const double PI = acos(-1.0);
const double eps = 1e-8;
const int maxn = 2000 + 10;
const int mod = 1e9 + 7;

int n;
char color[maxn];
std::vector<int> g[maxn];

int x[maxn][maxn], y[maxn][maxn];
int siz[maxn];
int dp1[maxn], dp2[maxn];

bool cmp(int u, int v) {
	return siz[u] < siz[v];
}

void dfs_size(int u, int f) {
	siz[u] = 1;
	
	for (int i = 0; i < g[u].size(); ++i) {
		int v = g[u][i];
		if (v != f) {
			dfs_size(v, u);
			siz[u] += siz[v];
		}
	}
	return;
}

void dfs(int u, int f) {
	fill(x[u], x[u] + siz[u] + 1, -inf);
	fill(y[u], y[u] + siz[u] + 1, inf);
	
	x[u][0] = y[u][0] = 0;
	x[u][1] = y[u][1] = (color[u] == '1' ? 1 : 0);
	int m = 1;
	
	sort(g[u].begin(), g[u].end(), cmp);
	
	for (int i = 0; i < g[u].size(); ++i) {
		int v = g[u][i];
		if (v != f) {
			dfs(v, u);
			
			for (int j = m; j > 0; j--) {
				for (int k = siz[v]; k > 0; k--) {
					x[u][j + k] = max(x[u][j + k], x[u][j] + x[v][k]);
					y[u][j + k] = min(y[u][j + k], y[u][j] + y[v][k]);
				}
			}
			m += siz[v];
		}
	}
}

int main() {
#ifndef ONLINE_JUDGE
	freopen("test.in", "r", stdin);
#endif
	
	int T;
	scanf("%d", &T);
	while (T--) {
		scanf("%d", &n);
		for (int i = 0; i <= n; ++i) {
			g[i].clear();
		}
		
		scanf("%s", color + 1);
		for (int i = 0; i < n - 1; ++i) {
			int u, v;
			scanf("%d%d", &u, &v);
			g[u].push_back(v);
			g[v].push_back(u);
		}
		
		dfs_size(1, 0);
		dfs(1, 0);
		
		fill(dp1, dp1 + n + 1, -inf);
		fill(dp2, dp2 + n + 1, inf);
		for (int i = 1; i <= n; i++) {
			for (int j = 0; j <= siz[i]; ++j) {
				dp1[j] = max(dp1[j], x[i][j]);
				dp2[j] = min(dp2[j], y[i][j]);
			}
		}
		
		
		ll ans = 0;
		for (int i = 0; i <= n; i++) {
			for (int j = dp2[i]; j <= dp1[i]; j++) {
				ans += (j + 1) * (i - j + 1);
			}
		}
		printf("%lld\n", ans);
		
	}
	
}