結果

問題 No.196 典型DP (1)
ユーザー Div9851Div9851
提出日時 2017-10-08 00:29:25
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 73 ms / 2,000 ms
コード長 1,197 bytes
コンパイル時間 1,289 ms
コンパイル使用メモリ 161,624 KB
実行使用メモリ 97,464 KB
最終ジャッジ日時 2024-11-17 04:41:33
合計ジャッジ時間 3,396 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

#include "bits/stdc++.h"
using namespace std;
#define rep(i,n) for(int i=0;i<n;i++)
#define ALL(v) (v).begin(),(v).end()
#define MP(a,b) make_pair(a,b)
typedef long long LL;
typedef pair<int, int> PI;
typedef vector<int> VI;
const LL MOD = 1000000007LL;
vector<int> G[2000];
LL dp[2000][2001];
int tree[2000];
int N, K;
void cnt(int v, int par) {
	tree[v] = 1;
	rep(i, G[v].size()) {
		int to = G[v][i];
		if (to == par) continue;
		cnt(to, v);
		tree[v] += tree[to];
	}
}
void dfs(int v, int par) {
	LL dp2[2][2001] = { 0 };
	int sum = 0;
	int now = 0;
	dp2[now][0] = 1;
	rep(i, G[v].size()) {
		int to = G[v][i];
		if (to == par) continue;
		dfs(to, v);
		for (int j = 0; j <= sum; j++) {
			for (int k = 0; k <= tree[to]; k++) {
				LL t = (dp2[now][j] * dp[to][k]) % MOD;
				if (j + k <= K) (dp2[now^1][j + k] += t) %= MOD;
			}
		}
		for (int j = 0; j <= tree[v]; j++) dp2[now][j] = 0;
		sum += tree[to];
		now ^= 1;
	}
	(dp2[now][tree[v]] += 1) %= MOD;
	for (int i = 0; i <= tree[v]; i++) dp[v][i] = dp2[now][i];
}
int main() {
	cin >> N >> K;
	rep(i, N - 1) {
		int a, b;
		cin >> a >> b;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	cnt(0, -1);
	dfs(0, -1);
	cout << dp[0][K] << endl;

}
0