codeforces1249F最大重量サブセット

トピックリンク：CF

ここに提示され、\（O（N ^ 3） \） で練習dpが

注\は（F_ {U、I} \） を表し（U \）\サブツリーをルート、およびすべての選択された点\（U \）少なくともの距離\（I \）最大点と右

\（I = 0 \） 、マストはグループから選択される\（U \）残りのすべての息子であり、\（V \）で\（F_ {V、K} \）

\は、（i> 0 \） 、我々は列挙（\ DIS）\を最小サブツリー点である\（V \） 、回答の列挙サブツリーツリー知ら寄与である（F_ {V、I-を\します1} \） 、サブツリーの残りの部分を考慮した後（ルートと呼ばれる\（W \）この時点での中ではないので）、より\）\（V浅い点で選択されたサブツリー、それらがにありますルート\（W \） A \（DIS \）定数（\ \ GEQ-W 1 \）の任意の2点間で、プラスに少なくとも距離\（K + 1 \）ので、サブツリー\（W \）貢献\（F_ {W、最大（KI ,. 1-I）} \） 。最後に、すべての\は（V \）回答取る\（最大\を）。

また、上記の私たちが知る必要があるということであることに注意してください\（のF \）状態に応じて、サフィックス最高を求めてこの非常に明確に定義されました

最終的な答えは（F_ {1,0} \）\します

 #include<iostream>
#include<string.h>
#include<string>
#include<stdio.h>
#include<algorithm>
#include<vector>
#include<math.h>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
typedef long double db;
typedef pair<int,int> pii;
const int N=10000;
const db pi=acos(-1.0);
#define lowbit(x) (x)&(-x)
#define sqr(x) (x)*(x)
#define rep(i,a,b) for (register int i=a;i<=b;i++)
#define per(i,a,b) for (register int i=a;i>=b;i--)
#define fir first
#define sec second
#define mp(a,b) make_pair(a,b)
#define pb(a) push_back(a)
#define maxd 998244353
#define eps 1e-8
struct node{int to,nxt;}sq[420];
int all=0,head[220];
int n,lim,a[220],f[220][220];

int read()
{
    int x=0,f=1;char ch=getchar();
    while ((ch<'0') || (ch>'9')) {if (ch=='-') f=-1;ch=getchar();}
    while ((ch>='0') && (ch<='9')) {x=x*10+(ch-'0');ch=getchar();}
    return x*f;
}

void add(int u,int v)
{
    all++;sq[all].to=v;sq[all].nxt=head[u];head[u]=all;
}

void dfs(int u,int fu)
{
    for (int i=head[u];i;i=sq[i].nxt)
    {
        int v=sq[i].to;
        if (v==fu) continue;
        dfs(v,u);
    }
    rep(j,0,lim)
    {
        if (j==0)
        {
            f[u][0]=a[u];
            for (int i=head[u];i;i=sq[i].nxt)
            {
                int v=sq[i].to;
                if (v!=fu) f[u][0]+=f[v][lim];
            }
        }
        else
        {
            for (int i=head[u];i;i=sq[i].nxt)
            {
                int v=sq[i].to,now=f[v][j-1];
                if (v==fu) continue;
                for (int k=head[u];k;k=sq[k].nxt)
                {
                    int w=sq[k].to;
                    if ((v==w) || (fu==w)) continue;
                    now+=f[w][max(lim-j,j-1)];
                }
                f[u][j]=max(f[u][j],now);
            }
        }
    }
    per(i,lim,0) f[u][i]=max(f[u][i],f[u][i+1]);
}

int main()
{
    n=read();lim=read();
    rep(i,1,n) a[i]=read();
    rep(i,1,n-1)
    {
        int u=read(),v=read();
        add(u,v);add(v,u);
    }
    dfs(1,0);
    printf("%d",f[1][0]);
    return 0;
}