【CF161D】Distance in Tree

链接：洛谷CF161D，有中文题面

解法：树形dp

k⩽100 k ⩽ 100 看着就像树形dp的范围，用 fu,d f u , d 表示 u u 子树内与点 u u 距离为 d d 的点的个数，转移为

fu,d=∑fav=ufv,d−1 f u , d = ∑ f a v = u f v , d − 1

边界为 fu,0=1 f u , 0 = 1 。

代码

#include
#include
#include

using namespace std;

int n,x,y,k;
vector<int> point[50001];
long long tot,f[50001][501],ans;

void dfs(int u,int fa){
    f[u][0]=1ll;
    for(int v:point[u])if(v!=fa){dfs(v,u);for(int i=1;i<=k;++i)f[u][i]+=f[v][i-1];}
    ans+=f[u][k];tot=0ll;
    for(int v:point[u])if(v!=fa)for(int i=1;i1]*(f[u][k-i]-f[v][k-i-1]);ans+=tot>>1;
}

int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;iscanf("%d%d",&x,&y),point[x].push_back(y),point[y].push_back(x);
    dfs(1,0);
    printf("%lld",ans);
}