CF161D Distance in Tree

题目大意

输入点数为$N$一棵树

求树上长度恰好为$K$的路径个数

题解

　　据说正解点分

　　然而我用树形dp的，简单易懂

　　虽然跑得没有点分快就是了

//minamoto
#include<iostream>
#include<cstdio>
#define ll long long
using namespace std;
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
inline int read(){
    #define num ch-'0'
    char ch;bool flag=0;int res;
    while(!isdigit(ch=getc()))
    (ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getc());res=res*10+num);
    (flag)&&(res=-res);
    #undef num
    return res;
}
const int N=50005;
int ver[N<<1],head[N],Next[N<<1];
ll res,f[N][505],ans;int n,k,tot;
inline void add(int u,int v){
    ver[++tot]=v,Next[tot]=head[u],head[u]=tot;
    ver[++tot]=u,Next[tot]=head[v],head[v]=tot;
}
void dfs(int u,int fa){
    f[u][0]=1;
    for(int i=head[u];i;i=Next[i]){
        int v=ver[i];if(v==fa) continue;dfs(v,u);
        for(int j=1;j<=k;++j) f[u][j]+=f[v][j-1];
    }
    ans+=f[u][k],res=0;
    for(int i=head[u];i;i=Next[i]){
        int v=ver[i];if(v==fa) continue;
        for(int j=1;j<k;++j) res+=f[v][j-1]*(f[u][k-j]-f[v][k-j-1]);
    }
    ans+=res>>1;
}
int main(){
    n=read(),k=read();
    for(int i=1;i<n;++i){
        int u=read(),v=read();
        add(u,v);
    }
    dfs(1,0);
    printf("%lld
",ans);
    return 0;
}