結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
26,068 KB
testcase_01 AC 6 ms
19,028 KB
testcase_02 AC 5 ms
19,004 KB
testcase_03 AC 5 ms
19,032 KB
testcase_04 AC 5 ms
19,004 KB
testcase_05 AC 5 ms
19,188 KB
testcase_06 AC 5 ms
19,096 KB
testcase_07 AC 4 ms
19,036 KB
testcase_08 AC 6 ms
19,120 KB
testcase_09 AC 5 ms
18,968 KB
testcase_10 AC 5 ms
18,856 KB
testcase_11 AC 5 ms
19,168 KB
testcase_12 AC 5 ms
19,072 KB
testcase_13 AC 5 ms
19,060 KB
testcase_14 AC 7 ms
19,112 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,296 KB
testcase_27 AC 9 ms
19,296 KB
testcase_28 AC 12 ms
19,244 KB
testcase_29 AC 16 ms
19,404 KB
testcase_30 AC 17 ms
19,480 KB
testcase_31 AC 5 ms
18,912 KB
testcase_32 AC 662 ms
19,152 KB
testcase_33 AC 1,663 ms
19,220 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; 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