結果

問題 No.196 典型DP (1)
ユーザー beet
提出日時 2018-06-21 19:39:28
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 844 bytes
コンパイル時間 1,674 ms
コンパイル使用メモリ 175,184 KB
実行使用メモリ 30,108 KB
最終ジャッジ日時 2024-06-30 17:40:05
合計ジャッジ時間 7,684 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 15 WA * 11 TLE * 1 -- * 14
権限があれば一括ダウンロードができます

ソースコード

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(sz[v]+1,0);
      dp[v][0]=1;
      for(Int u:G[v]){
	if(u==p) continue;
	vector<Int> nx(sz[v]+1,0);
	for(Int i=0;i<=sz[v];i++)
	  for(Int j=0;j<=sz[u];j++)
	    if(i+j<=sz[v]) nx[i+j]+=dp[v][i]*dp[u][j]%MOD;
	swap(dp[v],nx);
	for(Int i=0;i<=sz[v];i++) nx[i]%=MOD;
      }
      dp[v][sz[v]]=1;
    };
  dfs(0,-1);
  cout<<dp[0][k]<<endl;
  return 0;
}
0