結果

問題 No.196 典型DP (1)
ユーザー startcpp
提出日時 2018-03-07 20:45:17
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,567 bytes
コンパイル時間 651 ms
コンパイル使用メモリ 65,504 KB
実行使用メモリ 37,080 KB
最終ジャッジ日時 2024-10-04 09:01:18
合計ジャッジ時間 7,137 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 11 WA * 7 TLE * 2 -- * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

//0個以上の部分木を刈り取り、頂点数N-Kの木を作る。最終的にできる木は何通り？
//O(N^4)っぽい解法。
#include <iostream>
#include <vector>
#define int long long
#define rep(i, n) for(i = 0; i < n; i++)
using namespace std;

int mod = 1000000007;
int N, K;
vector<int> et[2000];
int treeSize[2000];

int sizeDfs(int p, int v) {
	int i, ret = 1;
	rep(i, et[v].size()) {
		if (et[v][i] == p) continue;
		ret += sizeDfs(v, et[v][i]);
	}
	return treeSize[v] = ret;
}

int treeDp[2000][2001];
//根の処理(v == 0の処理)に注意
int dfs(int p, int v, int c) {
	if (c > treeSize[v]) return 0;
	if (et[v].size() - (v != 0) <= 1) return 1;
	if (c <= 1) return 1;
	if (treeDp[v][c] != -1) return treeDp[v][c];
	
	int i, j, k;
	vector<vector<int>> dp;	//dp[i][j] = 子i個で合計j頂点残す方法の個数
	
	dp.resize(et[v].size() + (v == 0));
	rep(i, et[v].size() + (v == 0)) {
		dp[i].resize(c);
		for (j = 0; j < c; j++) {
			dp[i][j] = 0;
		}
	}
	dp[0][0] = 1;
	
	int chId = 0;
	rep(i, et[v].size()) {
		if (et[v][i] == p) continue;
		int ch = et[v][i];
		rep(j, c) {
			rep(k, treeSize[ch] + 1) {
				if (j + k >= c) break;
				dp[chId + 1][j + k] += dp[chId][j] * dfs(v, ch, k);
				dp[chId + 1][j + k] %= mod;
			}
		}
		chId++;
	}
	return treeDp[v][c] = dp[chId][c - 1];
}

signed main() {
	int i, j;
	
	cin >> N >> K;
	rep(i, N) rep(j, N + 1) treeDp[i][j] = -1;
	rep(i, N - 1) {
		int a, b;
		cin >> a >> b;
		et[a].push_back(b);
		et[b].push_back(a);
	}
	sizeDfs(-1, 0);
	cout << dfs(-1, 0, N - K) << endl;
	return 0;
}
0