Codeforces 1249 F Maximum Weight Subset —— 树形dp

This way

题意：

给你一棵树，让你在其中找出一个集合，集合中的点的任意两个点的距离大于k，并且所有点的权值和最大，问你这个值是多少。

题解：

怎么感觉以前做过。。
首先很明显是树形dp，那么对于当前点，有取和不取两种情况，那么对于取的情况，剩下的点一定是在比当前点的深度大k以下的地方，不取的时候，我们就又要考虑最高的那个的深度，所以我们就需要二维的dp来维护，dp[i][j]表示当前点为i的时候，最浅的取的点的深度比i点大j的时候的答案最大是多少。
那么状态转移方程，就是枚举深度的同时，枚举所有儿子在最高点的情况，那么剩下一些点的深度一定是距离这个点k+1，并且要不小于当前点的深度。
注意的一点就是要将低的值转移到高的地方，因为高的地方包含低的地方：

for(int i=k-1;~i;i--)dp[x][i]=max(dp[x][i],dp[x][i+1]);

#include<bits/stdc++.h>
using namespace std;
const int N=205;
struct node
{
    int to,next;
}e[N*2];
int cnt,head[N];
void add(int x,int y){
    e[cnt].to=y;
    e[cnt].next=head[x];
    head[x]=cnt++;
}
int a[N],dp[N][N],ans,k;
void dfs(int x,int fa){
    dp[x][0]=a[x];
    int sum[N]={0};
    for(int i=head[x];~i;i=e[i].next){
        int ne=e[i].to;
        if(ne==fa)continue;
        dfs(ne,x);
        dp[x][0]+=dp[ne][k];
        for(int j=0;j<=k;j++)
            sum[j]+=dp[ne][j];
    }
    ans=max(ans,dp[x][0]);
    for(int i=1;i<=k;i++){
        for(int j=head[x];~j;j=e[j].next){
            int ne=e[j].to;
            if(ne==fa)continue;
            dp[x][i]=max(dp[x][i],dp[ne][i-1]+sum[max(k-i,i-1)]-dp[ne][max(k-i,i-1)]);
            ans=max(ans,dp[x][i]);
        }
    }
    for(int i=k-1;~i;i--)dp[x][i]=max(dp[x][i],dp[x][i+1]);
}
int main()
{
    int n,x,y;
    scanf("%d%d",&n,&k);
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<n;i++)scanf("%d%d",&x,&y),add(x,y),add(y,x);
    dfs(1,0);
    printf("%d\n",ans);
    return 0;
}