結果
問題 | No.196 典型DP (1) |
ユーザー | Today03 |
提出日時 | 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 |
ソースコード
#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; }