Codeforces 161D 树形DP

题意：给你一颗数，边权值为1，问有多少对节点满足最小距离为k (n <= 50000 , 1 <= k <= 500)

由于k很小，所以我们可以直接对于每个节点保存该节点子树中到它距离为dis(0 <= dis <= k)的点数，对于一个节点，每处理出新一个子树，计算下新子树里的点到已处理的子树里的点满足条件的种数，然后将新子树距离加到已处理的。。

#include <stdio.h>
#include <string.h>

struct EDGE {
int to, next;
}edge[100005];
struct PP {
int dis[505];
};
int E, head[50005], ans ,K;
void newedge(int u, int to) {
edge[E].to = to;
edge[E].next = head[u];
head[u] = E++;
}
void init() {
memset(head, -1, sizeof(head));
E = 0;
ans = 0;
}
PP dfs(int u, int pre) {
PP a , b;
memset(a.dis, 0, sizeof(a.dis));
if(head[u] == -1) {
a.dis[0] = 1;
return a;
}
int i, j;
for(i = head[u];i != -1; i = edge[i].next) {
int to = edge[i].to;
if(to == pre) continue;
b = dfs(to, u);
ans += b.dis[K-1];
for(j = 1;j < K; j++)
ans += a.dis[j-1]*b.dis[K-j-1];
for(j = 0;j < K; j++)
a.dis[j] += b.dis[j];
}
for(i = K;i > 0; i--)
a.dis[i] = a.dis[i-1];
a.dis[0] = 1;
return a;
}
int main() {
int n, a, b, i;
while(scanf("%d%d", &n, &K) != -1) {
init();
for(i = 0;i < n-1; i++) {
scanf("%d%d", &a, &b);
newedge(a, b);
newedge(b, a);
}
dfs(1, -1);
printf("%d\n", ans);
}
return 0;
}