結果

問題 No.196 典型DP (1)
ユーザー Div9851Div9851
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0