結果

問題 No.196 典型DP (1)
ユーザー evimaevima
提出日時 2015-03-14 10:16:44
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 12 ms / 2,000 ms
コード長 1,421 bytes
コンパイル時間 1,319 ms
コンパイル使用メモリ 164,648 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-28 22:53:33
合計ジャッジ時間 3,025 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

// Enjoy your stay.

#include <bits/stdc++.h>

#define long long long
#define LOOPVAR_TYPE long

#define all(x) (x).begin(), (x).end()
#define sz(x) ((LOOPVAR_TYPE)(x).size())
#define foreach(it, X) for(__typeof((X).begin()) it = (X).begin(); it != (X).end(); it++)
#define GET_MACRO(_1, _2, _3, NAME, ...) NAME
#define _rep(i, n) _rep2(i, 0, n)
#define _rep2(i, a, b) for(LOOPVAR_TYPE i = (LOOPVAR_TYPE)(a); i < (LOOPVAR_TYPE)(b); i++)
#define rep(...) GET_MACRO(__VA_ARGS__, _rep2, _rep)(__VA_ARGS__)

#define fir first
#define sec second
#define mp make_pair
#define mt make_tuple
#define pb push_back

const double EPS = 1e-9;
const double PI = acos(-1.0);
const long INF = 1070000000LL;
const long MOD = 1000000007LL;

using namespace std;

typedef istringstream iss;
typedef stringstream sst;
typedef pair<LOOPVAR_TYPE, LOOPVAR_TYPE> pi;
typedef vector<LOOPVAR_TYPE> vi;

const int MN = 2010;

int N, K;
vi g[MN];

vi f(int cur, int pre){
	vi res(1, 1);
	for(int ch: g[cur]){
		if(ch == pre) continue;
		vi sub = f(ch, cur);
		int n = sz(res) - 1;
		int m = sz(sub) - 1;
		vi nres(n + m + 1);
		rep(i, n+1){
			rep(j, m+1){
				nres[i + j] = (nres[i + j] + res[i] * sub[j]) % MOD;
			}
		}
		res = nres;
	}
	res.pb(1);
	return res;
}

int main(){
	cin.tie(0);
	ios_base::sync_with_stdio(0);
	
	cin >> N >> K;
	rep(i, N-1){
		int a, b;
		cin >> a >> b;
		g[a].pb(b);
		g[b].pb(a);
	}
	cout << f(0, -1)[K] << endl;
}
0