LibreOJ #2473. 「九省联考 2018」秘密袭击 树形依赖背包

题意

给出一棵树，每个节点有一个权值。问所有连通块中第k大元素的和模64123。
设最权值为$W$,$n\le1666,W\le1666$

分析

正解什么的当然是不会啦，那我们考虑如何暴力。
只要枚举每个点，然后以这个点为根做一次树形依赖背包就好了。
做完之后发现并不能跑过去。
考虑优化，当不小于点x的节点数小于k时，就可以直接退出啦。
然后你会发现，你不仅AC了，而且跑的比标程还要快！

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=2005;
const int MOD=64123;

int n,k,w,f[N][N],last[N],cnt,a[N],val[N];
struct edge{int to,next;}e[N*2];

void mod(int &x)
{
    x-=x>=MOD?MOD:0;
}

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

void dfs(int x,int fa)
{
    if (val[x])
    {
        for (int i=k;i>=1;i--) f[x][i]=f[x][i-1];
        f[x][0]=0;
    }
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa) continue;
        for (int j=0;j<=k;j++) f[e[i].to][j]=f[x][j];
        dfs(e[i].to,x);
        for (int j=0;j<=k;j++) mod(f[x][j]+=f[e[i].to][j]);
    }
}

int main()
{
    scanf("%d%d%d",&n,&k,&w);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    for (int i=1;i<n;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        addedge(x,y);
    }
    int ans=0;
    for (int i=1;i<=n;i++)
    {
        int tot=0;
        for (int j=1;j<=n;j++) val[j]=a[j]>a[i]||a[j]==a[i]&&j>=i,tot+=val[j];
        if (tot<k) continue;
        for (int j=0;j<=k;j++) f[i][j]=0;
        f[i][0]=1;
        dfs(i,0);
        (ans+=f[i][k]*a[i])%=MOD;
    }
    printf("%d",ans);
    return 0;
}