【南外】DPS1-A

发布时间 2023-09-15 20:36:48作者: Binary_1110011
/*
树形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;
}