/*
树形DP,记 f[i][j] 表示第 i 个节点,其子树中包含它的长度为 j 的链的数量。
对于一个节点 x,其有子节点 V1,V2,...
设当前做到 Vi,则 lst[i][...] 表示做到 Vi-1 时 f[x][...] 的值。
统计答案时,对于新增点 Vi,假设我们已经跟新完了 f(即 f[x][j] 加上 f[Vi][j-1]),
则答案的增加量为 Σ(1<=j<=k) (f[x][j]-lst[x][j])*lst[x][k-j]
意为(新增加进来的子节点中长度为 j 的链数量)*(原有的长度为 k-j 的链数量)
然后更新 lst 即可。
*/
//If, one day, I finally manage to make my dreams a reality...
//I wonder, will you still be there by my side?
#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(false)
#define TIE cin.tie(0),cout.tie(0)
#define int long long
#define y1 cyy
#define fi first
#define se second
#define cnt1(x) __builtin_popcount(x)
#define mk make_pair
#define pb push_back
#define pii pair<int,int>
#define ls(x) (x<<1)
#define rs(x) (x<<1|1)
#define lbt(x) (x&(-x))
using namespace std;
int n,k,u,v,f[50005][505],lst[50005][505],ans;
vector<int> a[200005];
void dfs(int x,int fa){
f[x][0]=lst[x][0]=1;
for(auto tmp:a[x]){
if(tmp!=fa){
dfs(tmp,x);
for(int i=1;i<=k;i++) f[x][i]+=f[tmp][i-1];
for(int i=k;i>=1;i--){
ans+=(f[x][i]-lst[x][i])*lst[x][k-i];
}
for(int i=1;i<=k;i++) lst[x][i]+=f[tmp][i-1];
}
}
}
signed main(){
IOS;TIE;
cin>>n>>k;
for(int i=1;i<n;i++){
cin>>u>>v;
a[u].pb(v);
a[v].pb(u);
}
dfs(1,0);
cout<<ans<<endl;
return 0;
}
标签:DPS1,tmp,int,Vi,dfs,南外,lst,define
From: https://www.cnblogs.com/binary1110011/p/17705893.html