結果

問題 No.196 典型DP (1)
ユーザー kurarrrkurarrr
提出日時 2018-02-21 10:57:12
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 36 ms / 2,000 ms
コード長 1,531 bytes
コンパイル時間 1,517 ms
コンパイル使用メモリ 163,816 KB
実行使用メモリ 35,072 KB
最終ジャッジ日時 2024-09-14 15:14:11
合計ジャッジ時間 3,559 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"

#define ALL(g) (g).begin(),(g).end()
#define REP(i, x, n) for(int i = x; i < n; i++)
#define rep(i,n) REP(i,0,n)
#define RREP(i, x, n) for(int i = x; i >= n; i--)
#define rrep(i, n) RREP(i,n,0)
#define pb push_back

template<class T> void chmax(T& a,const T& b){a=max(a,b);}
template<class T> void chmin(T& a,const T& b){a=min(a,b);}

using namespace std;

using ll = long long;
using P = pair<int,int>;
using Pl = pair<ll,ll>;
using vi = vector<int>;
using vvi = vector<vi>;

const int mod=1e9+7,INF=1<<30;
const double EPS=1e-12,PI=3.1415926535897932384626;
const ll lmod = 1e9+7,LINF=1LL<<60;
const int MAX_N = 2003;

ll dp[MAX_N][MAX_N];
vector<int> graph[MAX_N];
int part[MAX_N];

int main(){
  int N,K;
  cin >> N >> K;
  rep(i,N-1){
    int a,b; cin >> a >> b;
    graph[a].pb(b); graph[b].pb(a);
  }
  rep(i,N) rep(j,K+1) dp[i][j] = 0LL;
  rep(i,N) dp[i][0] = 1LL;
  fill(part,part+N,1);
  function<void(int,int)> dfs = [&](int u,int p){
    for(auto v:graph[u]){
      if(v==p) continue;
      dfs(v,u);
      part[u] += part[v];
      // 元々の部分木 -> part[u]-part[v]-1
      // 新しい部分木 -> part[v]
      vector<ll> table(part[u]+1,0);
      rep(j,part[u]-part[v]){
        rep(l,part[v]+1){
          (table[j+l] += dp[u][j] * dp[v][l]) %= lmod;
        }
      }
      rep(j,part[u]) dp[u][j] = table[j];
    }
    dp[u][part[u]] = 1LL;
    return ;
  };
  dfs(0,-1);
  cout << dp[0][K] << endl;
  // rep(i,N){ rep(j,K+1) cout << dp[i][j] << " "; cout << endl;}
  return 0;
}
0