結果

問題 No.196 典型DP (1)
コンテスト
ユーザー ゴリポン先生
提出日時 2026-01-30 19:28:57
言語 D
(dmd 2.111.0)
結果
AC  
実行時間 27 ms / 2,000 ms
コード長 793 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,169 ms
コンパイル使用メモリ 188,112 KB
実行使用メモリ 7,972 KB
最終ジャッジ日時 2026-01-30 19:29:03
合計ジャッジ時間 5,561 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

module main;
// https://mayokoex.hatenablog.com/entry/2015/05/03/122625 より
// 木DP、深さ優先探索
import std;

int N, K;
int[][] G;
immutable MOD = 1_000_000_007L;

long[] dfs(int v, int p)
{
	long[] ret = [1L];
	foreach (next; G[v]) {
		if (next == p) continue;
		long[] tmp = dfs(next, v);
		int n = ret.length.to!int;
		int m = tmp.length.to!int;
		auto nret = new long[](n + m - 1);
		foreach (i; 0 .. n) foreach (j; 0 .. m) {
			(nret[i + j] += ret[i] * tmp[j]) %= MOD;
		}
		ret = nret;
	}
	ret ~= 1;
	return ret;
}

void main()
{
	// 入力
	readln.chomp.formattedRead("%d %d", N, K);
	G = new int[][](N);
	foreach (_; 0 .. N - 1) {
		int a, b;
		readln.chomp.formattedRead("%d %d", a, b);
		G[a] ~= b;
		G[b] ~= a;
	}
	// 答えの計算と出力
	writeln(dfs(0, -1)[K]);
}
0