結果

問題 No.196 典型DP (1)
ユーザー Kmcode1
提出日時 2015-04-26 23:06:34
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,621 bytes
コンパイル時間 1,303 ms
コンパイル使用メモリ 108,880 KB
実行使用メモリ 16,256 KB
最終ジャッジ日時 2024-07-05 01:08:42
合計ジャッジ時間 7,448 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 7 WA * 11 TLE * 1 -- * 22
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:75:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   75 |         scanf("%d%d", &n, &k);
      |         ~~~~~^~~~~~~~~~~~~~~~
main.cpp:78:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   78 |                 scanf("%d%d", &a, &b);
      |                 ~~~~~^~~~~~~~~~~~~~~~

ソースコード

diff # 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cctype>
#include<cstdlib>
#include<algorithm>
#include<bitset>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<map>
#include<set>
#include<stack>
#include<cmath>
#include<sstream>
#include<fstream>
#include<iomanip>
#include<ctime>
#include<complex>
#include<functional>
#include<climits>
#include<cassert>
#include<iterator>
#include<valarray>
//#include<bits/stdc++.h>
//using namespace std;
using namespace std;
#define MOD 1000000007

int n;
int k;
#define MAX 2001
vector<int> g[MAX];
bool use[MAX];
vector<int> v[MAX];
int countt[MAX];
inline void dfs(int b){
	use[b] = true;
	for (int i = 0; i < v[b].size(); i++){
		if (use[v[b][i]] == false){
			dfs(v[b][i]);
			g[b].push_back(v[b][i]);
		}
	}
}
inline void dfs2(int b){
	countt[b] = 1;
	for (int i = 0; i < g[b].size(); i++){
		dfs2(g[b][i]);
		countt[b] += countt[g[b][i]];
	}
}
long long int dp[MAX][MAX];
inline void dfs3(int b){
	for (int i = 0; i < g[b].size(); i++){
		dfs3(g[b][i]);
	}
	dp[b][0] = 1;
	dp[b][countt[b]] = 1;
	for (int i = 0; i < g[b].size(); i++){
		int jyo = min(k, countt[g[b][i]]);
		int jj = min(jyo + 1, k);
		for (int j = k/*jj*/; j >= 0; j--){
			//choose from g[b][i]
			for (int kk = 1; kk <=k/*<= jyo&&kk <= j*/; kk++){
				dp[b][j] += dp[g[b][i]][kk] * dp[b][j - kk];
				dp[b][j] %= MOD;
			}
		}
	}
}
int main(){
	scanf("%d%d", &n, &k);
	for (int i = 1; i < n; i++){
		int a, b;
		scanf("%d%d", &a, &b);
		v[a].push_back(b);
		v[b].push_back(a);
	}
	dfs(0);
	dfs2(0);
	dfs3(0);
	printf("%lld\n", dp[0][k]);
	return 0;
}
0