結果
問題 | No.196 典型DP (1) |
ユーザー | ctyl_0 |
提出日時 | 2015-10-05 19:43:28 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 605 ms / 2,000 ms |
コード長 | 1,794 bytes |
コンパイル時間 | 1,703 ms |
コンパイル使用メモリ | 93,228 KB |
実行使用メモリ | 222,976 KB |
最終ジャッジ日時 | 2024-07-20 01:11:40 |
合計ジャッジ時間 | 14,445 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <iostream> #include <iomanip> #include <vector> #include <algorithm> #include <numeric> #include <functional> #include <cmath> #include <queue> #include <stack> #include <set> #include <map> #define repd(i,a,b) for (int i=(a);i<(b);i++) #define rep(i,n) repd(i,0,n) #define var auto #define mod 1000000007 typedef long long ll; using namespace std; int inputValue(){ int a; cin >> a; return a; }; template <typename T> void output(T a, int precision) { if(precision > 0){ cout << setprecision(precision) << a << "\n"; } else{ cout << a << "\n"; } } vector<ll> dnc(vector<vector<int>> ngb, int n, int p){ // 親がpの頂点nを根にしたsubtreeで0~n個塗る場合の数を返す vector<ll> ret(1, 1); // 初期値: 0個塗る場合の数は1通り rep(i, ngb[n].size()){ if (ngb[n][i] == p) { // 今いる頂点の親と子(実は親)が一緒の場合は何もしない continue; } // 分割統治 vector<ll> add = dnc(ngb, ngb[n][i], n); vector<ll> new_ret(ret.size() + add.size() - 1, 0); rep(j, ret.size()){ rep(k, add.size()){ new_ret[j + k] = (new_ret[j + k] + ret[j] * add[k]) % mod; } } ret = new_ret; } ret.push_back(1); // 頂点nを塗る場合は全て塗るので1通り return ret; } int main() { // source code int N = inputValue(); int K = inputValue(); vector<vector<int>> ngb(N); // neighbor rep(i, N - 1){ int a = inputValue(); int b = inputValue(); ngb[a].push_back(b); ngb[b].push_back(a); } var ans = dnc(ngb, 0, -1); output(dnc(ngb, 0, -1)[K], 0); return 0; }