結果

問題 No.196 典型DP (1)
ユーザー pekempey
提出日時 2015-10-23 17:18:53
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 24 ms / 2,000 ms
コード長 1,272 bytes
コンパイル時間 1,405 ms
コンパイル使用メモリ 161,832 KB
実行使用メモリ 26,496 KB
最終ジャッジ日時 2024-07-22 11:41:58
合計ジャッジ時間 3,229 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define GET_MACRO(a, b, c, NAME, ...) NAME
#define rep(...) GET_MACRO(__VA_ARGS__, rep3, rep2)(__VA_ARGS__)
#define rep2(i, a) rep3 (i, 0, a)
#define rep3(i, a, b) for (int i = (a); i < (b); i++)
#define repr(...) GET_MACRO(__VA_ARGS__, repr3, repr2)(__VA_ARGS__)
#define repr2(i, a) repr3 (i, 0, a)
#define repr3(i, a, b) for (int i = (b) - 1; i >= (a); i--)
#define chmin(a, b) ((b) < a && (a = (b), true))
#define chmax(a, b) (a < (b) && (a = (b), true))
using namespace std;
typedef long long ll;

vector<int> G[2020];
ll dp[2020][2020];
int size[2020];
const ll mod = 1e9 + 7;

int dfs2(int curr, int prev) {
	size[curr] = 1;
	for (int next : G[curr]) if (next != prev) {
		size[curr] += dfs2(next, curr);
	}
	return size[curr];
}

void dfs(int curr, int prev) {
	dp[curr][0] = 1;
	dp[curr][size[curr]] = 1;
	for (int next : G[curr]) if (next != prev) {
		dfs(next, curr);
		repr (i, size[curr]) if (dp[curr][i] > 0) rep (j, 1, size[next] + 1) {
			if (i + j <= size[curr])
				(dp[curr][i + j] += dp[curr][i] * dp[next][j]) %= mod;
		}
	}
}

int main() {
	int N, K;
	cin >> N >> K;
	rep (i, N - 1) {
		int u, v;
		cin >> u >> v;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs2(0, -1);
	dfs(0, -1);
	cout << dp[0][K] << endl;
	return 0;
}
0