結果

問題 No.196 典型DP (1)
ユーザー sugim48sugim48
提出日時 2015-04-26 22:32:00
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 54 ms / 2,000 ms
コード長 1,437 bytes
コンパイル時間 876 ms
コンパイル使用メモリ 99,296 KB
実行使用メモリ 11,904 KB
最終ジャッジ日時 2024-07-05 03:01:25
合計ジャッジ時間 3,017 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

#define _USE_MATH_DEFINES
#include <algorithm>
#include <cstdio>
#include <functional>
#include <iostream>
#include <cfloat>
#include <climits>
#include <cstring>
#include <cmath>
#include <fstream>
#include <map>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <time.h>
#include <vector>
using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> i_i;
typedef pair<ll, int> ll_i;
typedef pair<double, int> d_i;
typedef pair<ll, ll> ll_ll;
typedef pair<double, double> d_d;
struct edge { int u, v; ll w; };

int INF = INT_MAX / 10;
ll MOD = 1000000007;
ll _MOD = 1000000009;
double EPS = 1e-10;

void f(int u, int p, vector<vector<int> >& G, vector<vector<int> >& a) {
	a[u] = vector<int>(2, 0);
	a[u][1] = 1;
	for (int i = 0; i < G[u].size(); i++) {
		int v = G[u][i];
		if (v == p) continue;
		f(v, u, G, a);
		int n = a[u].size(), m = a[v].size();
		vector<int> b(min(2001, n + m - 1));
		for (int x = 0; x < n; x++)
			for (int y = 0; y < m; y++)
				if (x + y <= 2000)
					b[x + y] = (b[x + y] + (ll)a[u][x] * a[v][y]) % MOD;
		a[u] = b;
	}
	a[u][0] = (a[u][0] + 1) % MOD;
}

int main() {
	int N, K; cin >> N >> K;
	vector<vector<int> > G(N);
	for (int i = 0; i < N - 1; i++) {
		int a, b; cin >> a >> b;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	vector<vector<int> > a(N);
	f(0, -1, G, a);
	cout << a[0][N - K] << endl;
}
0