結果

問題 No.196 典型DP (1)
ユーザー kurarrrkurarrr
提出日時 2018-02-21 10:57:12
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 36 ms / 2,000 ms
コード長 1,531 bytes
コンパイル時間 1,517 ms
コンパイル使用メモリ 163,816 KB
実行使用メモリ 35,072 KB
最終ジャッジ日時 2024-09-14 15:14:11
合計ジャッジ時間 3,559 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 3 ms
5,632 KB
testcase_16 AC 7 ms
8,960 KB
testcase_17 AC 10 ms
12,544 KB
testcase_18 AC 16 ms
19,072 KB
testcase_19 AC 15 ms
17,280 KB
testcase_20 AC 13 ms
12,544 KB
testcase_21 AC 14 ms
13,952 KB
testcase_22 AC 13 ms
12,160 KB
testcase_23 AC 20 ms
20,224 KB
testcase_24 AC 17 ms
16,256 KB
testcase_25 AC 26 ms
28,544 KB
testcase_26 AC 24 ms
26,496 KB
testcase_27 AC 25 ms
26,624 KB
testcase_28 AC 30 ms
30,080 KB
testcase_29 AC 34 ms
35,072 KB
testcase_30 AC 33 ms
34,944 KB
testcase_31 AC 18 ms
11,648 KB
testcase_32 AC 25 ms
20,992 KB
testcase_33 AC 30 ms
27,136 KB
testcase_34 AC 36 ms
34,816 KB
testcase_35 AC 36 ms
34,944 KB
testcase_36 AC 18 ms
11,392 KB
testcase_37 AC 20 ms
20,864 KB
testcase_38 AC 29 ms
27,136 KB
testcase_39 AC 31 ms
29,568 KB
testcase_40 AC 35 ms
34,816 KB
testcase_41 AC 2 ms
5,376 KB
testcase_42 AC 2 ms
5,376 KB
testcase_43 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"

#define ALL(g) (g).begin(),(g).end()
#define REP(i, x, n) for(int i = x; i < n; i++)
#define rep(i,n) REP(i,0,n)
#define RREP(i, x, n) for(int i = x; i >= n; i--)
#define rrep(i, n) RREP(i,n,0)
#define pb push_back

template<class T> void chmax(T& a,const T& b){a=max(a,b);}
template<class T> void chmin(T& a,const T& b){a=min(a,b);}

using namespace std;

using ll = long long;
using P = pair<int,int>;
using Pl = pair<ll,ll>;
using vi = vector<int>;
using vvi = vector<vi>;

const int mod=1e9+7,INF=1<<30;
const double EPS=1e-12,PI=3.1415926535897932384626;
const ll lmod = 1e9+7,LINF=1LL<<60;
const int MAX_N = 2003;

ll dp[MAX_N][MAX_N];
vector<int> graph[MAX_N];
int part[MAX_N];

int main(){
  int N,K;
  cin >> N >> K;
  rep(i,N-1){
    int a,b; cin >> a >> b;
    graph[a].pb(b); graph[b].pb(a);
  }
  rep(i,N) rep(j,K+1) dp[i][j] = 0LL;
  rep(i,N) dp[i][0] = 1LL;
  fill(part,part+N,1);
  function<void(int,int)> dfs = [&](int u,int p){
    for(auto v:graph[u]){
      if(v==p) continue;
      dfs(v,u);
      part[u] += part[v];
      // 元々の部分木 -> part[u]-part[v]-1
      // 新しい部分木 -> part[v]
      vector<ll> table(part[u]+1,0);
      rep(j,part[u]-part[v]){
        rep(l,part[v]+1){
          (table[j+l] += dp[u][j] * dp[v][l]) %= lmod;
        }
      }
      rep(j,part[u]) dp[u][j] = table[j];
    }
    dp[u][part[u]] = 1LL;
    return ;
  };
  dfs(0,-1);
  cout << dp[0][K] << endl;
  // rep(i,N){ rep(j,K+1) cout << dp[i][j] << " "; cout << endl;}
  return 0;
}
0