結果
問題 | No.196 典型DP (1) |
ユーザー | kazuma |
提出日時 | 2018-11-27 13:20:00 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 25 ms / 2,000 ms |
コード長 | 1,990 bytes |
コンパイル時間 | 1,819 ms |
コンパイル使用メモリ | 169,868 KB |
実行使用メモリ | 19,584 KB |
最終ジャッジ日時 | 2024-06-23 08:41:22 |
合計ジャッジ時間 | 3,744 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h>using namespace std;template<int MOD>class mod_int {unsigned x;public:mod_int() : x(0) { }mod_int(int sig) { int sigt = sig % MOD; if (sigt < 0) sigt += MOD; x = sigt; }mod_int(long long sig) { int sigt = sig % MOD; if (sigt < 0) sigt += MOD; x = sigt; }int get() const { return (int)x; }mod_int &operator+=(mod_int that) { if ((x += that.x) >= MOD) x -= MOD; return *this; }mod_int &operator-=(mod_int that) { if ((x += MOD - that.x) >= MOD) x -= MOD; return *this; }mod_int &operator*=(mod_int that) { x = (unsigned long long)x * that.x % MOD; return *this; }mod_int &operator/=(mod_int that) { return *this *= that.inverse(); }mod_int operator+(mod_int that) const { return mod_int(*this) += that; }mod_int operator-(mod_int that) const { return mod_int(*this) -= that; }mod_int operator*(mod_int that) const { return mod_int(*this) *= that; }mod_int operator/(mod_int that) const { return mod_int(*this) /= that; }bool operator==(const mod_int& that) const { return x == that.x; }mod_int inverse() const {long long a = x, b = MOD, u = 1, v = 0;while (b) {long long t = a / b;a -= t * b; swap(a, b);u -= t * v; swap(u, v);}return mod_int(u);}};const int mod = 1e9 + 7;using mint = mod_int<mod>;const int MAX = 2000;vector<int> G[MAX];int size[MAX];mint dp[MAX][MAX + 1];mint tmp[MAX + 1];void dfs(int v, int prev) {size[v] = 0;dp[v][0] = 1;for (auto to : G[v]) if (to != prev) {dfs(to, v);for (int j = 0; j <= size[v]; j++) {tmp[j] = dp[v][j];dp[v][j] = 0;}for (int j = size[v]; j >= 0; j--) {for (int k = 0; k <= size[to]; k++) {dp[v][j + k] += tmp[j] * dp[to][k];}}size[v] += size[to];}size[v] += 1;dp[v][size[v]] += 1;}int main(){int N, K;cin >> N >> K;for (int i = 0; i < N - 1; i++) {int u, v;cin >> u >> v;G[u].push_back(v);G[v].push_back(u);}dfs(0, -1);cout << dp[0][K].get() << endl;return 0;}