結果
問題 | 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;}