結果

問題 No.196 典型DP (1)
ユーザー beetbeet
提出日時 2018-06-21 19:47:09
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 32 ms / 2,000 ms
コード長 842 bytes
コンパイル時間 1,698 ms
コンパイル使用メモリ 174,592 KB
実行使用メモリ 34,176 KB
最終ジャッジ日時 2024-06-30 17:39:57
合計ジャッジ時間 3,367 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
//INSERT ABOVE HERE
signed main(){
  Int n,k;
  cin>>n>>k;
  vector<vector<Int> > G(n);
  for(Int i=1;i<n;i++){
    Int x,y;
    cin>>x>>y;
    G[x].emplace_back(y);
    G[y].emplace_back(x);
  }
  const Int MOD = 1e9+7;
  vector<vector<Int> > dp(n);
  vector<Int> sz(n,1);
  function<void(Int,Int)> dfs=
    [&](Int v,Int p){
      for(Int u:G[v]){
	if(u==p) continue;
	dfs(u,v);
	sz[v]+=sz[u];
      }
      dp[v].assign(1,1);
      for(Int u:G[v]){
	if(u==p) continue;
	vector<Int> nx(dp[v].size()+sz[u],0);
	for(Int i=0;i<(Int)dp[v].size();i++)
	  for(Int j=0;j<=sz[u];j++)
	    nx[i+j]+=dp[v][i]*dp[u][j]%MOD;
	for(Int i=0;i<(Int)nx.size();i++) nx[i]%=MOD;
	dp[v]=move(nx);
      }
      dp[v].emplace_back(1);
    };
  dfs(0,-1);
  cout<<dp[0][k]<<endl;
  return 0;
}
0