codeforces 161D - Distance in Tree 树状DP

题目给出一棵树，求任意两点间距离为k的节点的个数

方法是：深搜遍历整棵树的同时，对于每一个节点，每求出一个以其为根的子树中任意节点到其的距离数组后，先由乘法原理维护最终答案，再维护其遍历过的子树到其距离数组

乘法原理，即，因为k可由(1, k-1) (2, k-2).....(k - 1, 1)构成（有顺序）， 所以其所由可能为p[i] * q[k - i] (1 <= i < k)  p,q分别为遍历过的子树的距离数组，当前遍历的子树的距离数组

 

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <vector>
using namespace std;
vector<int> e[55555];
int dp[55555][555];
int n,k;
int dfs(int root,int fa,int& ans)
{
	int dep=0;
	dp[root][0]=1;
	for(int i=0;i<e[root].size();i++)
	{
		if(e[root][i]==fa) continue;
		dep=max(dfs(e[root][i],root,ans),dep);
		if(dep>k) dep=k;
		for(int j=1;j<=k;j++)
		{
			ans+=dp[root][j-1]*dp[e[root][i]][k-j];//更新答案
		}
		for(int j=1;j<=dep;j++)
		{
			dp[root][j]+=dp[e[root][i]][j-1];//维护新的dp数组
		}
	}
	return dep+1;
}
int main()
{
	#ifndef ONLINE_JUDGE
		freopen("G:/1.txt","r",stdin);
		freopen("G:/2.txt","w",stdout);
	#endif

	scanf("%d%d",&n,&k);
	int tmp1,tmp2;
	for(int i=0;i<n-1;i++)
	{
		scanf("%d%d",&tmp1,&tmp2);
		e[tmp1].push_back(tmp2);
		e[tmp2].push_back(tmp1);
	}
	int ans=0;
	dfs(1,-1,ans);
	printf("%d\n",ans);
	return 0;
}