A - Distance in Tree

題意：給你一棵樹，讓你記錄樹上兩點距離為k的點對數。
思路：
dp[i][j]代表考慮i這顆子樹，與它距離為j的點數量，dp[i][0]就是1了，然後dfs處理一下就可以，那麼它就是答案的一部分，還有一部分就是把i當做中轉節點，從u開始遞迴，列舉它的每個子樹，ans+=(dp[u][tt]-dp[j][tt-1])*(dp[j][k-tt-1])，最後別忘記除以二，最後有點小疑惑，為什麼這樣就是答案了呢，會不會重複或者漏算呢，首先dp[i][k]加起來肯定是不重複的，然後畫畫圖發現把u當做中轉的也不會有重複。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=50010,M=100010;
int n,k;
int h[N],e[M],ne[M];
int dp[N][510];
ll ans;
int idx;
void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int fa)
{
	dp[u][0]=1;
	for(int i=h[u];~i;i=ne[i])
	{
		int j=e[i];
		if(j==fa)
		continue;
		dfs(j,u);
		for(int t=1;t<=k;t++)
		dp[u][t]+=dp[j][t-1];
	}
	ans+=dp[u][k];
	ll tmp=0;
	for(int i=h[u];~i;i=ne[i])
	{
		int j=e[i];
		if(j==fa)	
		continue;
	for(int t=1;k-t-1>=0;t++)
		tmp+=(dp[u][t]-dp[j][t-1])*(dp[j][k-t-1]);
		//ans+=tmp/2;
	}
	ans+=tmp/2;
}
int main()
{
	cin>>n>>k;
	memset(h,-1,sizeof h);
	for(int i=0;i<n-1;i++)
	{
		int a,b;
		cin>>a>>b;
		add(a,b);
		add(b,a);
	}
	dfs(1,-1);
//	for(int i=1;i<=n;i++)
//	{
//	for(int j=0;j<=k;j++)
//	cout<<dp[i][j]<<' ';
//	cout<<endl;
//	}
	
	cout<<ans<<endl;
	return 0;
}