1097G Vladislav and a Great Legend

传送门

分析

https://blog.csdn.net/forever_shi/article/details/88048528

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
#define int long long
const int mod = 1e9+;
int dp[][],now[],siz[],n,k;
int s[][],p[],res[];
vector<int>v[];
inline void init(){
    int i,j;
    p[]=;
    for(i=;i<=k;i++)p[i]=p[i-]*i%mod;
    s[][]=;
    for(i=;i<=k;i++)
      for(j=;j<=i;j++)
        s[i][j]=(s[i-][j-]+s[i-][j]*j%mod)%mod;
}
inline void dfs(int x,int fa){
    dp[x][]=;
    for(int i=;i<v[x].size();i++){
      if(v[x][i]==fa)continue;
      dfs(v[x][i],x);
      for(int j=;j<=k;j++)now[j]=dp[x][j];
      for(int j=;j<=k;j++)dp[x][j]=(dp[x][j]+dp[v[x][i]][j]+dp[v[x][i]][j-])%mod;
      dp[x][]=(dp[x][]+dp[v[x][i]][])%mod;
      for(int j=;j<=min(k,siz[x]);j++)
        for(int t=;t<=min(k-j,siz[v[x][i]]);t++){
          dp[x][j+t]=(dp[x][j+t]+now[j]*dp[v[x][i]][t]%mod)%mod;
          dp[x][j+t+]=(dp[x][j+t+]+now[j]*dp[v[x][i]][t]%mod)%mod;
          res[j+t]=(res[j+t]+now[j]*dp[v[x][i]][t]%mod)%mod;
          res[j+t+]=(res[j+t+]+now[j]*dp[v[x][i]][t]%mod)%mod;
        }
      siz[x]+=siz[v[x][i]]+;
    }
}
signed main(){
    int i,j,Ans=;
    scanf("%lld%lld",&n,&k);
    for(i=;i<n;i++){
      int x,y;
      scanf("%lld%lld",&x,&y);
      v[x].push_back(y);
      v[y].push_back(x);
    }
    init();
    dfs(,);
    for(i=;i<=k;i++)
      Ans=(Ans+p[i]*s[k][i]%mod*res[i]%mod)%mod;
      cout<<Ans;
    return ;
}