結果

問題 No.196 典型DP (1)
コンテスト
ユーザー koba-e964
提出日時 2015-06-05 17:35:59
言語 C++11（廃止可能性あり）
(gcc 15.2.0 + boost 1.89.0)
結果
RE  
実行時間 -
コード長 1,758 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 762 ms
コンパイル使用メモリ 91,344 KB
実行使用メモリ 814,080 KB
最終ジャッジ日時 2024-07-06 14:14:35
合計ジャッジ時間 6,282 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other RE * 14 MLE * 1 -- * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int rec_p(int, int)’:
main.cpp:54:1: warning: no return statement in function returning non-void [-Wreturn-type]
   54 | }
      | ^

ソースコード

diff #
raw source code 

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <deque>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <utility>
#include <vector>

#define REP(i,s,n) for(int i=(int)(s);i<(int)(n);i++)

using namespace std;
typedef long long int ll;
typedef vector<int> VI;
typedef pair<int, int> PI;
const double EPS=1e-9;

const int N = 2010;

const int DEBUG = 0;
VI graph[N];
int parent[N];
int desc[N];

ll dp[N][N];
const ll mod = 1e9+7;
int rec_p(int n, int p) {
  parent[n] = -2;
  int s = 0;
  REP(i, 0, graph[n].size()) {
    int v = graph[n][i];
    if (parent[v] != -2) {
      rec_p(v, n);
      s += desc[v] + 1;
    }
  }
  parent[n] = p;
  desc[n] = s;
}
// [0,d]
void poly_mul(ll aa[N], const ll bb[N], int d) {
  ll cc[N] = {0};
  REP(i, 0, d + 1) {
    REP(j, 0, d - i + 1) {
      cc[i + j] += aa[i] * bb[j];
      cc[i + j] %= mod;
    }
  }
  REP(i, 0, d + 1) {
    aa[i] = cc[i];
  }
}

void rec(int v) {
  REP(i, 0, graph[v].size()) {
    int u = graph[v][i];
    if (u == parent[v]) {
      continue;
    }
    rec(u);
    poly_mul(dp[v], dp[u], desc[v]);
  }
  dp[v][desc[v] + 1] = 1;
}

int main(void){
  int n, k;
  cin >> n >> k;
  REP (i, 0, n - 1) {
    int a, b;
    cin >> a >> b;
    graph[a].push_back(b);
    graph[b].push_back(a);
  }
  rec_p(0, -1);
  if (DEBUG) {
    REP(i, 0, n) {
      cout << "p[" << i << "]=" << parent[i] << endl;
    }
  }
  REP(i,0,N) {
    fill_n(dp[i], N, 0);
    dp[i][0] = 1;
  }
  rec(0);
  cout << dp[0][k] << endl;
}
0