codeforces 791D 树形dp 所有点对

题意：20w个点的树，边长都为1，每个点可以往距离不超过k<=5的点进行一次跳跃。求所有的f(A,B)的和，f(A,B)是A到B所需的最小跳跃次数，标号A

题解：树形dp，A

#include 
using namespace std;
typedef long long int ll;
ll dp[250000][5];
vectorg[250000];
int k;
ll dists[220000],ans,sons[220000];
void dfs(int x,int fa)
{
	sons[x]=1;
	dp[x][0]=1;
	int u;
	for(int t=0;t=k)ans+=dp[x][i]*dp[u][j];
			}
        sons[x]+=sons[u];
        dists[x]+=dists[u];
        dists[x]+=dp[u][k-1];
            for(int i=0;i>n>>k;ans=0;
	for(int i=1;i>a>>b;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	dfs(1,0);
	cout<