結果

問題 No.196 典型DP (1)
ユーザー ei1333ei1333
提出日時 2015-07-08 17:42:31
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 1,101 bytes
コンパイル時間 1,227 ms
コンパイル使用メモリ 161,628 KB
実行使用メモリ 26,160 KB
最終ジャッジ日時 2024-07-08 01:43:54
合計ジャッジ時間 7,776 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
26,160 KB
testcase_01 AC 5 ms
19,148 KB
testcase_02 AC 4 ms
19,184 KB
testcase_03 AC 5 ms
19,100 KB
testcase_04 AC 4 ms
19,004 KB
testcase_05 AC 6 ms
19,124 KB
testcase_06 AC 5 ms
19,188 KB
testcase_07 AC 6 ms
18,968 KB
testcase_08 AC 5 ms
19,076 KB
testcase_09 AC 4 ms
19,116 KB
testcase_10 AC 7 ms
19,188 KB
testcase_11 AC 6 ms
19,056 KB
testcase_12 AC 5 ms
19,144 KB
testcase_13 AC 5 ms
19,076 KB
testcase_14 AC 6 ms
19,104 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 AC 8 ms
19,160 KB
testcase_27 AC 8 ms
19,364 KB
testcase_28 AC 12 ms
19,452 KB
testcase_29 AC 15 ms
19,356 KB
testcase_30 AC 17 ms
19,412 KB
testcase_31 AC 6 ms
19,196 KB
testcase_32 AC 659 ms
19,112 KB
testcase_33 AC 1,658 ms
19,240 KB
testcase_34 TLE -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
const int mod = 1000000007;

int N, K;
vector< int > graph[2000];
int dp[2000][2001];
int size[2000]; // SubTreeSize

int dfs(int idx, int prev)
{
  size[idx] = 1;
  for(int i = 0; i < graph[idx].size(); i++) {
    if(graph[idx][i] == prev) continue;
    size[idx] += dfs(graph[idx][i], idx);
  }
  return(size[idx]);
}


int rec(int idx, int prev, int mul)
{
  if(~dp[idx][mul]) return(dp[idx][mul]);
  int* dp2 = dp[idx];
  for(int i = 0; i <= K; i++) dp2[i] = 0;
  dp2[0] = 1;
  for(int i = 0; i < graph[idx].size(); i++) {
    if(graph[idx][i] == prev) continue;
    for(int j = K - 1; j >= 0; j--) {
      if(dp2[j] == 0) continue;
      for(int k = K - j; k > 0; k--) {
        (dp2[j + k] += dp2[j] * rec(graph[idx][i], idx, k)) %= mod;
      }
    }
  }
  (dp2[size[idx]] += 1) %= mod;
  return(dp2[mul]);
}

int main()
{ 
  fill_n(*dp, 2000 * 2001, -1);

  cin >> N >> K;
  for(int i = 0; i < N - 1; i++) {
    int a, b;
    cin >> a >> b;
    graph[a].push_back(b);
    graph[b].push_back(a);
  }
  dfs(0, -1);
  cout << rec(0, -1, K) << endl;
}
0