結果

問題 No.196 典型DP (1)
ユーザー autumn-eel
提出日時 2018-01-01 21:30:42
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 20 ms / 2,000 ms
コード長 1,621 bytes
コンパイル時間 1,978 ms
コンパイル使用メモリ 168,968 KB
実行使用メモリ 19,712 KB
最終ジャッジ日時 2024-12-22 23:55:21
合計ジャッジ時間 3,470 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n)for(int i=0;i<(n);i++)
#define MOD 1000000007
using namespace std;
typedef long long ll;

int dp[3000][3000];//頂点iの個をj個塗る通り数
vector<int>E[3000];
int siz[3000];//頂点iを根とする部分木のサイズ

void rec(int v,int p){
	dp[v][0]=1;
	siz[v]=1;
	for(auto u:E[v]){
		if(u==p)continue;
		rec(u,v);
		for(int i=siz[v];i>=0;i--){
			for(int j=1;j<=siz[u];j++){
				(dp[v][i+j]+=(dp[u][j]*(ll)dp[v][i])%MOD)%=MOD;
			}
		}
		siz[v]+=siz[u];
	}
	dp[v][siz[v]]=1;
}
int main(){
	int n,k;scanf("%d%d",&n,&k);
	rep(i,n-1){
		int a,b;scanf("%d%d",&a,&b);
		E[a].push_back(b);
		E[b].push_back(a);
	}
	rec(0,-1);
	cout<<dp[0][k]<<endl;
}/*
#include<bits/stdc++.h>
using namespace std;
const int mod = 1000000007;
int N, K;
vector< int > graph[2000];
int cnt=0;

int dp[2000][2001];
int num[2000];
int rec(int idx, int prev)
{
  for(int i = 0; i < graph[idx].size(); i++) {
    if(graph[idx][i] != prev) rec(graph[idx][i], idx);
  }
  dp[idx][0] = 1;
  num[idx] = 1;
  for(int i = 0; i < graph[idx].size(); i++) {
    if(graph[idx][i] == prev) continue;
    for(int j = num[idx]; j >= 0; j--) {
      for(int k = num[graph[idx][i]]; k >= 1; k--) {
        (dp[idx][j + k] += (long long)dp[idx][j] * dp[graph[idx][i]][k] % mod) %= mod;
        cnt++;
      }
    }
    num[idx] += num[graph[idx][i]];
  }
  dp[idx][num[idx]] = 1;
  return(dp[idx][K]);
}

int main()
{
  cin >> N >> K;
  for(int i = 0; i < N - 1; i++) {
    int a, b;
    cin >> a >> b;
    graph[a].push_back(b);
    graph[b].push_back(a);
  }
  cout << rec(0, -1) << endl;
  cout<<cnt<<endl;
}*/
0