bzoj5250 [2018多省省队联测]秘密袭击

时间:2022-04-14 21:47:26

博主蒟蒻,目前还不会动态dp,所以下面说的是一个并不优秀的暴力,我会补的!

我们考虑按权值从大到小依次点亮每个点,相同权值可以同时点亮,每次点亮后,我们进行一次树形背包。

处理出$f[i][j]$表示i的子树中有j个亮点的方案数,然后就AC了。

有两个小优化,一个是将背包的枚举上限设为min(size[x],K),此处size[x]为子树中点亮的点的的个数。

还有就是我们可以把大于K的dp值都和K合并到一起,因为我们需要的是所有大于等于K的方案数。

 #include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#define N 1700
#define mod 64123
using namespace std;
int e=,head[N];
struct edge{
int v,next;
}ed[N<<];
void add(int u,int v){
ed[e].v=v;
ed[e].next=head[u];
head[u]=e++;
}
int n,m,K,a[N],pp[N];
int f[N][N],g[N],size[N],vis[N],ans,sum,last;
bool cmp(int a,int b){return ::a[a]>::a[b];}
void dfs(int x,int fa){
for(int i=;i<=K;i++)f[x][i]=;
size[x]=vis[x];f[x][size[x]]=;
for(int i=head[x];i;i=ed[i].next){
int v=ed[i].v;
if(v==fa)continue;
dfs(v,x);
int up1=min(size[x],K),up2=min(size[v],K);
for(int j=min(up1+up2,K);~j;j--)g[j]=;
for(int j=up1;~j;j--)
for(int k=up2;~k;k--)
(g[min(j+k,K)]+=1ll*f[x][j]*f[v][k]%mod)%=mod;
size[x]+=size[v];
up1=min(size[x],K);
for(int j=;j<=up1;j++)f[x][j]=g[j];
}
f[x][]++;
(sum+=f[x][K])%=mod;
}
int main(){
scanf("%d%d%d",&n,&K,&m);
for(int i=;i<=n;i++){
scanf("%d",&a[i]);
pp[i]=i;
}
sort(pp+,pp+n+,cmp);
for(int i=,u,v;i<n;i++){
scanf("%d%d",&u,&v);
add(u,v);add(v,u);
}
for(int i=;i<=n;){
do vis[pp[i]]=,i++;
while(i<=n&&a[pp[i]]==a[pp[i-]]);
if(i<K)continue;
sum=;dfs(,);
(ans+=1ll*(sum-last+mod)*a[pp[i-]]%mod)%=mod;
last=sum;
}
printf("%d\n",ans);
return ;
}