牛客训练，杀树，树形dp

D杀树

一道树形dp题，我刚开始想的是$dp[i][j]$表示i为根结点最长链的长度为j的最小代价，转移的时候很复杂，最后还是没写出来（也有可能是写炸了，但是这个状态应该也是可以的）。比赛结束后看到题解是用$dp[i][j]$表示i为根结点最长链的长度小于等于j的最小代价，这样转移起来就很轻松了，所以做dp的时候选好状态真的很重要…
首先递归，让子节点的值全部更新完成。再更新当前结点的值。
递推式：
当前结点删除：当前代价加所有（子节点的最小代价）的和
$dp[now][0]=a[now]+{\sum }_{S=son(now)}dp[S][l-1]$
当前结点不删：注意以now为根的树的直径要小于l
一般的$dp[now][i]={\sum }_{S=son(now)}dp[S][min(i-1)]）$
对于直径大于l的：只能最多取一个子节点长度为i-1其余取l-i-1保证直径在l以下！

#include<cstdio>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<vector>
#include<functional>
using namespace std;
typedef long long LL;
inline LL read()
{
    LL kk=0,f=1;
    char cc=getchar();
    while(cc<'0'||cc>'9'){if(cc=='-')f=-1;cc=getchar();}
    while(cc>='0'&&cc<='9'){kk=(kk<<1)+(kk<<3)+cc-'0';cc=getchar();}
    return kk*f;
}
const LL maxn=5055;
LL n,l,a[maxn],dp[maxn][maxn];
vector<int>edge[maxn];
void add(int a,int b)
{
	edge[a].push_back(b);
	edge[b].push_back(a);
}
void dfs(int now,int fa)
{
	dp[now][0]=a[now];
	for(int i=0;i<edge[now].size();++i)
	{
		int to=edge[now][i];
		if(to!=fa)
		{
			dfs(to,now);
			dp[now][0]+=dp[to][l-1];//处理当前截断的代价
		}
	}
	for(LL i=1;i<l;++i)
	{
		LL tmp=0;
		for(int j=0;j<edge[now].size();++j)
		{
			int to=edge[now][j];
			if(to==fa)continue;
			dp[now][i]=
			min(tmp+dp[to][i-1],dp[now][i]+dp[to][min(i-1,l-i-1)]);//如果i-1>l-i-1,如果随便取，直径大于l，tmp记录之前没有取i-1的时候的最优解，tmp+dp[][]表示当前取i-1，后面的表示当前不取，两者取最优
		    tmp+=dp[to][min(i-1,l-i-1)];
		}
		dp[now][i]=min(dp[now][i],dp[now][i-1]);//保证dp[i][j]表示<=j
	}
}
int main()
{
	n=read();l=read();
	for(int i=1;i<=n;++i)a[i]=read();
	for(int i=1;i<n;++i)add(read(),read());
	dfs(1,0);
	printf("%lld\n",dp[1][l-1]);
}