結果
問題 | No.196 典型DP (1) |
ユーザー | Div9851 |
提出日時 | 2017-10-08 00:29:25 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 73 ms / 2,000 ms |
コード長 | 1,197 bytes |
コンパイル時間 | 1,289 ms |
コンパイル使用メモリ | 161,624 KB |
実行使用メモリ | 97,464 KB |
最終ジャッジ日時 | 2024-11-17 04:41:33 |
合計ジャッジ時間 | 3,396 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include "bits/stdc++.h" using namespace std; #define rep(i,n) for(int i=0;i<n;i++) #define ALL(v) (v).begin(),(v).end() #define MP(a,b) make_pair(a,b) typedef long long LL; typedef pair<int, int> PI; typedef vector<int> VI; const LL MOD = 1000000007LL; vector<int> G[2000]; LL dp[2000][2001]; int tree[2000]; int N, K; void cnt(int v, int par) { tree[v] = 1; rep(i, G[v].size()) { int to = G[v][i]; if (to == par) continue; cnt(to, v); tree[v] += tree[to]; } } void dfs(int v, int par) { LL dp2[2][2001] = { 0 }; int sum = 0; int now = 0; dp2[now][0] = 1; rep(i, G[v].size()) { int to = G[v][i]; if (to == par) continue; dfs(to, v); for (int j = 0; j <= sum; j++) { for (int k = 0; k <= tree[to]; k++) { LL t = (dp2[now][j] * dp[to][k]) % MOD; if (j + k <= K) (dp2[now^1][j + k] += t) %= MOD; } } for (int j = 0; j <= tree[v]; j++) dp2[now][j] = 0; sum += tree[to]; now ^= 1; } (dp2[now][tree[v]] += 1) %= MOD; for (int i = 0; i <= tree[v]; i++) dp[v][i] = dp2[now][i]; } int main() { cin >> N >> K; rep(i, N - 1) { int a, b; cin >> a >> b; G[a].push_back(b); G[b].push_back(a); } cnt(0, -1); dfs(0, -1); cout << dp[0][K] << endl; }