[BZOJ5250][树形DP]九省联考2018：秘密袭击

BZOJ5250

听说正解是FFT

不管反正他们都暴力A了，我也要暴力A
考虑每个点对ans的贡献，把点权大于它的点设为1，小于它的设为0，然后树上背包求树上有多少个有k个1的连通块，乘以a[i]就是贡献

Code:

#include<bits/stdc++.h>
#define ll long long
#define mod 64123
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=2005;
int n,k,w,s;
int ans=0;
int b[N];
int vis[N<<1],head[N<<1],nxt[N<<1],tot=0;
int dp[N][N];
inline void add(int x,int y) {vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;}
inline void _add(int &x,int y){x+=y;if(x>=mod) x-=mod;}
void dfs(int v,int fa){
    if(b[s]<b[v] || (b[v]==b[s] && v<s)) for(int i=1;i<k;i++) _add(dp[v][i+1],dp[fa][i]);
    else for(int i=1;i<=k;i++) _add(dp[v][i],dp[fa][i]);
    for(int i=head[v];i;i=nxt[i]) if(vis[i]!=fa) dfs(vis[i],v);
    if(fa==0) return;
    for(int i=1;i<=k;i++) _add(dp[fa][i],dp[v][i]);
}
int main(){
	n=read(),k=read(),w=read();
	for(int i=1;i<=n;i++) b[i]=read();
	for(int x,y,i=1;i<n;i++){
		x=read(),y=read();add(x,y);add(y,x);
	}
	for(int i=1;i<=n;i++){
		int cnt=0;
		for(int j=1;j<=n;j++) if(b[j]>b[i] || (b[j]==b[i] && i>j)) ++cnt;
		if(cnt<k-1) continue;
		memset(dp,0,sizeof(dp));
		dp[i][1]=1;
		s=i;
		dfs(i,0);
		ans=((ll)ans+(ll)b[i]*dp[i][k])%mod;
	}
	cout<<ans<<endl;
	return 0;
}