結果

問題 No.196 典型DP (1)
ユーザー Vi24EVi24E
提出日時 2024-02-07 01:22:06
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 18 ms / 2,000 ms
コード長 797 bytes
コンパイル時間 4,353 ms
コンパイル使用メモリ 269,456 KB
実行使用メモリ 12,032 KB
最終ジャッジ日時 2024-09-28 12:22:18
合計ジャッジ時間 5,945 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#include<atcoder/all>
using namespace atcoder;
vector<vector<modint1000000007>> dp(3030);
vector<vector<int>> g;

void solve(int t, int p){
	if (g[t].size() == 1 && p != -1){
		dp[t] = {1, 1};
		return;
	}

	dp[t] = {1, 0};
	for (auto x : g[t]){
		if (x == p) continue;
		solve(x, t);

		vector<modint1000000007> merge(dp[t].size() + dp[x].size() - 1);
		for (int i = 0; i < (int)dp[t].size(); i++){
			for (int j = 0; j < (int)dp[x].size(); j++){
				merge[i + j] += dp[t][i] * dp[x][j];
			}
		}
		dp[t] = merge;
	}
	dp[t].back() = 1;
}

int main(){
	int n, k;
	cin >> n >> k;
	g.resize(n);
	for (int i = 0; i < n - 1; i++){
		int u, v;
		cin >> u >> v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	
	solve(0, -1);
	cout << dp[0][k].val() << endl;
}
0