【ybtoj高效进阶 21188】树上赋值（树形DP）

树上赋值

题目链接：ybtoj高效进阶 21188

题目大意

给你一棵树，你要给每个点赋上 1~m 范围的一个值，然后保证任意一条边连着的两个点的权值差大于等于给出的 k，然后问你有多少种赋值方法。

思路

考虑先暴力 DP，设 $f_{i,j}$ 为处理好 $i$ 的子树，然后 $i$ 这个点选的 $j$ 这个值的方案数。

然后就枚举每个儿子，$f_{u,j}$ 乘上满足的 $l$ 的 $f_{v,l}$ 的和。（满足的 $l$ 是 $|j-l|⩾k$）

然后我们可以用前缀和处理一个区间的 $f_{x,i}$ 和，但是 $O(nm)$ 还是会 T。
然后你考虑 $m$ 那么大，中间很多部分都是相同的值。

那我们会发现它最大有不同的就在 $(n-1)\ast k$ 个。（两半各有那么多）
而且两边的还是对称的，所以我们就可以只维护两边不同的那 $(n-1)\ast k$ 个，复杂度就是 $O(n\ast nk)=O(n^{2}k)$ 了。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long
#define mo 1000000007

using namespace std;

struct node {
	int to, nxt;
}e[201];
int T, n, m, k, le[101], KK, x, y, lim;
ll f[101][20001], sum[101][20001];

void add(int x, int y) {
	e[++KK] = (node){y, le[x]}; le[x] = KK;
}

ll ksm(ll x, ll y) {
	ll re = 1;
	while (y) {
		if (y & 1) re = re * x % mo;
		x = x * x % mo;
		y >>= 1;
	}
	return re;
}

ll clac(int now, int r) {//算出前缀
	if (r <= lim) return sum[now][r];
	if (r <= m - lim) {
		return (sum[now][lim] + f[now][lim] * (r - lim) % mo) % mo;
	}
	return (sum[now][lim] + f[now][lim] * (m - 2 * lim) % mo + sum[now][lim] - sum[now][m - r] + mo) % mo;
}

void dfs(int now, int father) {
	for (int i = 1; i <= lim; i++)
		f[now][i] = 1;
	for (int i = le[now]; i; i = e[i].nxt)
		if (e[i].to != father) {
			dfs(e[i].to, now);
			for (int j = 1; j <= lim; j++) {
				int L = max(0, j - k);
				ll val = sum[e[i].to][L];
				int R = min(m, j + k - 1);
				val = (val + clac(e[i].to, m) - clac(e[i].to, R) + mo) % mo; 
				f[now][j] = f[now][j] * val % mo;
			}
		}
	for (int i = 1; i <= lim; i++)
		sum[now][i] = (sum[now][i - 1] + f[now][i]) % mo;
}

int main() {
//	freopen("label.in", "r", stdin);
//	freopen("label.out", "w", stdout);
	
	scanf("%d", &T);
	while (T--) {
		scanf("%d %d %d", &n, &m, &k);
		for (int i = 1; i <= n; i++) le[i] = 0; KK = 0;
		for (int i = 1; i < n; i++) {
			scanf("%d %d", &x, &y); add(x, y); add(y, x);
		}
		
		if (!k) {
			printf("%lld\n", ksm(m, n));
			continue;
		}
		
		lim = (n - 1) * k;
		if (2 * lim >= m) lim = m;
		
		dfs(1, 0);
		
		printf("%lld\n", clac(1, m));
	}
	
	return 0;
}