結果

問題 No.196 典型DP (1)
ユーザー Vi24E
提出日時 2024-02-07 01:14:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 773 bytes
コンパイル時間 3,948 ms
コンパイル使用メモリ 257,364 KB
最終ジャッジ日時 2025-02-19 02:40:13
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample RE * 3
other RE * 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};
	}

	dp[t] = {1};
	for (auto x : g[t]){
		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[t][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;
		u--; v--;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	
	solve(0, -1);
	cout << dp[0][k].val() << endl;
}
0