結果

問題 No.196 典型DP (1)
ユーザー Today03Today03
提出日時 2023-05-07 23:19:51
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 11 ms / 2,000 ms
コード長 887 bytes
コンパイル時間 2,158 ms
コンパイル使用メモリ 208,380 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-24 21:50:10
合計ジャッジ時間 3,510 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

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

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 1000000007;
int main() {
int n, k;
cin >> n >> k;
vector<vector<int>> g(n);
for (int i = 0; i < n - 1; i++) {
int a, b;
cin >> a >> b;
g[a].push_back(b);
g[b].push_back(a);
}
auto merge = [&](const vector<long long> &a, const vector<long long> &b) {
vector<long long> ret(a.size() + b.size() - 1);
for (int i = 0; i < a.size(); i++) {
for (int j = 0; j < b.size(); j++) {
(ret[i + j] += a[i] * b[j]) %= MOD;
}
}
return ret;
};
auto dfs = [&](auto &&dfs, int now, int pre) -> vector<long long> {
vector<long long> dp(2);
dp[0] = 1;
for (int nex : g[now]) {
if (nex != pre) {
dp = merge(dp, dfs(dfs, nex, now));
}
}
dp.back() = 1;
return dp;
};
auto dp = dfs(dfs, 0, 0);
cout << dp[k] % MOD << endl;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0