【BZOJ5250】[九省联考2018]秘密袭击（动态规划）

题面

BZOJ
洛谷
给定一棵树，求其所有联通块的权值第\(k\)大的和。

题解

整个\(O(nk(n-k))\)的暴力剪剪枝就给过了。。。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define MAX 1700
#define MOD 64123
inline int read()
{
    int x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
struct Line{int v,next;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int n,K,W,ans,a[MAX],b[MAX];
int f[MAX][MAX],sz[MAX],tmp[MAX];
void dfs(int u,int ff)
{
    f[u][b[u]]=1;sz[u]=b[u];
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;if(v==ff)continue;
        dfs(v,u);
        for(int j=0;j<=sz[u]&&j<=K;++j)
            for(int k=0;k<=sz[v]&&k<=K;++k)
                tmp[j+k]=(tmp[j+k]+1ll*f[u][j]*f[v][k]%MOD);
        sz[u]+=sz[v];
        for(int j=0;j<=sz[u];++j)f[u][j]=(f[u][j]+tmp[j])%MOD,tmp[j]=0;
    }
}
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 u=read(),v=read();
        Add(u,v);Add(v,u);
    }
    for(int i=1;i<=n;++i)
    {
        int sum=0;
        for(int j=1;j<=n;++j)
            if(a[j]>a[i]||(a[j]==a[i]&&j>=i))b[j]=1,++sum;
            else b[j]=0;
        if(sum<K)continue;
        memset(f,0,sizeof(f));
        dfs(i,0);
        ans=(ans+1ll*a[i]*f[i][K])%MOD;
    }
    printf("%d\n",ans);
    return 0;
}

标签：ch,int,sum,九省,2018,include,联考,getchar
来源： https://www.cnblogs.com/cjyyb/p/10367187.html