結果

問題 No.196 典型DP (1)
ユーザー startcppstartcpp
提出日時 2018-03-07 20:45:17
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 1,567 bytes
コンパイル時間 731 ms
コンパイル使用メモリ 65,496 KB
実行使用メモリ 36,812 KB
最終ジャッジ日時 2024-04-15 01:35:12
合計ジャッジ時間 7,396 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
10,752 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 1 ms
6,940 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 WA -
testcase_11 AC 3 ms
6,944 KB
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 TLE -
testcase_19 TLE -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
権限があれば一括ダウンロードができます

ソースコード

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