树型dp hdu5593 ZYB's Tree

传送门：点击打开链接

题意：告诉你如何构造一颗树，然后询问有多少点对的距离小于等于K(K<=10)

思路：用csy的话来说，这就是个傻逼题，然而比赛的时候就是傻逼不会- -

设dp[u][k]表示当节点u作为子树的根节点时，在这个子树中有多少点对与u的距离<=k

那么ans[u]=dp[u][k]+sigma(dp[v][k-i]-dp[vlast][k-i-1]) 1<=i<=k，i表示向父节点走的距离，v表示距离u为i的父节点，vlast表示距离u为k-1的父节点。

其实就是说，如果选u作为一个点对的其中一个点后，另一个点要么出现在它作为根节点所在的子树中，那么这部分答案就是dp[u][k]

另外有可能就是在除了这个子树的其他位置上。那么我就去向上找父节点，以父节点作为拐弯点，求出dp[v][k-i]的个数，因为又包括了自己子树的一部分，再减去dp[vlast][k-i-1]就搞定了。

只有一个要注意的地方，就是生成树的那个函数中，A*i可能会爆int如果不用LL可能会RE，还有就是前面那个公式i=k的时候，要特判一下，因为此时k-i-1会是负数

#include<map>
#include<set>
#include<cmath>
#include<ctime>
#include<stack&