[九省联考2018]秘密袭击coat [推导+树形DP]

传送门

题意：求出树上一个大于等于 k 的连通块的 第 k 大之和

首先可以枚举第k大x，看它能在哪些集合中

然后就是一步比较经典的转换

显然第 x 大会被算 x 次

我们考虑哪些集合满足第 k 大 >= x

记 >= x 的数有 cnt 个，那么 cnt >= k 说明第 k 大>=x

于是我们考虑 DP 满足这个条件的连通块个数

然后就可以通过卡常技巧通过这道题

---

#include<bits/stdc++.h>
#define N 2005
using namespace std;
int read(){
	int cnt = 0; char ch = 0;
	while(!isdigit(ch)) ch = getchar();
	while(isdigit(ch)) cnt = (cnt<<1) + (cnt<<3) + (ch-'0'), ch = getchar();
	return cnt;
}
int first[N], nxt[N<<1], to[N<<1], tot;
void addedge(int x, int y){
	nxt[++tot] = first[x], first[x] = tot, to[tot] = y;
}
int n, k, w, a[N];
int f[N][N], ok[N], siz[N], ans;
const int Mod = 64123;
int add(int a, int b){ return (a + b > Mod) ? a + b - Mod : a + b;}
int mul(int a, int b){ return 1ll * a * b % Mod;}
void dfs(int u, int fa){
	siz[u] = ok[u];
	for(int i = 0; i <= k; i++) f[u][i] = 0;
	f[u][ok[u]] = 1;
	for(int e=first[u]; e; e=nxt[e]){
		int t = to[e]; if(t == fa) continue;
		dfs(t, u);
		for(int i = siz[u]; i >= 0; --i){
			int tmp = f[u][i];
			for(int j = siz[t]; j >= 0; --j){
				f[u][min(k, i + j)] = add(f[u][min(k, i + j)], mul(tmp, f[t][j]));
			}
		} siz[u] = min(siz[u] + siz[t], k);
	} 
}
int main(){
	n = read(), k = read(), w = read();
	for(int i=1; i<=n; i++) a[i] = read();
	for(int i=1; i<n; i++){
		int x = read(), y = read();
		addedge(x, y); addedge(y, x);
	} 
	for(int i=1; i<=w; i++){
		int cnt = 0;
		for(int j=1; j<=n; j++){
			if(a[j] >= i) cnt++, ok[j] = 1;
			else ok[j] = 0; 
		}
		if(cnt < k) break;
		dfs(1, 0);
		for(int j=1; j<=n; j++){
			ans = add(ans, f[j][k]);
		}
	} printf("%d", ans); return 0;
}