結果

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

ソースコード

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

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