Codeforces 1499 F. Diameter Cuts —— 树形DP，上下界优化

This way

题意：

给你一棵树，你要删掉一些边使得剩下的森林中所有树的直径不超过k，问你有多少种删除方法。

题解：

跟南京的那一场有点像？看到这个数据范围就想到树形dp+上下界优化了啊，这里有一点不同的是儿子转移到父亲的时候，直径判断是i+j+1<=k，但是传到父亲的时候，应当是max(i,j+1)，也就是最深的距离。应当很容易理解。
dp[i][j]表示到了第i个点，子树中最深的距离为j的时候的情况数。
注意一下转移的时候有两种状态：连这个子树和不连这个子树。不连的话，用加法原理将子树所有情况加起来乘到当前点上即可。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=5e3+5;
const ll mod=998244353;
ll dp[N][N],tmp[N];
int n,k,mx[N];
vector<int>vec[N];
void dfs(int x,int fa){
    dp[x][0]=1;
    mx[x]=0;
    for(int ne:vec[x])
        if(ne!=fa)
            dfs(ne,x);
    for(int ne:vec[x]){
        if(ne==fa)continue;
        for(int i=0;i<=min(max(mx[x],mx[ne]+1),k);i++)tmp[i]=0;
        ll cas=0;
        for(int i=0;i<=mx[ne];i++)cas=(cas+dp[ne][i])%mod;
        for(int i=0;i<=mx[x];i++)
            tmp[i]=dp[x][i]*cas%mod;
        for(int i=0;i<=mx[x];i++)
            for(int j=0;j<=mx[ne]&&i+j+1<=k;j++)
                tmp[max(i,j+1)]=(tmp[max(i,j+1)]+dp[x][i]*dp[ne][j])%mod;
        mx[x]=min(k,max(mx[x],mx[ne]+1));
        for(int i=0;i<=mx[x];i++)dp[x][i]=tmp[i];
    }
}
int main()
{
    scanf("%d%d",&n,&k);
    int x,y;
    for(int i=1;i<n;i++)
        scanf("%d%d",&x,&y),vec[x].push_back(y),vec[y].push_back(x);
    dfs(1,0);
    ll ans=0;
    for(int i=0;i<=k;i++)ans=(ans+dp[1][i])%mod;
    printf("%lld\n",ans);
    return 0;
}