結果

問題 No.196 典型DP (1)
ユーザー sune232002sune232002
提出日時 2015-04-26 23:44:56
言語 C++11
(gcc 11.4.0)
結果
TLE  
実行時間 -
コード長 4,332 bytes
コンパイル時間 1,420 ms
コンパイル使用メモリ 163,128 KB
実行使用メモリ 41,784 KB
最終ジャッジ日時 2024-07-05 02:56:07
合計ジャッジ時間 8,660 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 15 ms
41,660 KB
testcase_01 AC 15 ms
34,716 KB
testcase_02 AC 16 ms
34,592 KB
testcase_03 AC 15 ms
34,720 KB
testcase_04 AC 15 ms
34,720 KB
testcase_05 AC 16 ms
34,848 KB
testcase_06 AC 15 ms
34,716 KB
testcase_07 AC 15 ms
34,720 KB
testcase_08 AC 15 ms
34,716 KB
testcase_09 AC 15 ms
34,724 KB
testcase_10 AC 15 ms
34,720 KB
testcase_11 AC 16 ms
34,716 KB
testcase_12 AC 15 ms
34,720 KB
testcase_13 AC 14 ms
34,720 KB
testcase_14 AC 16 ms
34,716 KB
testcase_15 AC 18 ms
34,716 KB
testcase_16 AC 19 ms
34,592 KB
testcase_17 AC 31 ms
34,716 KB
testcase_18 AC 72 ms
34,720 KB
testcase_19 AC 76 ms
34,848 KB
testcase_20 AC 67 ms
34,724 KB
testcase_21 AC 92 ms
34,720 KB
testcase_22 AC 102 ms
34,724 KB
testcase_23 AC 1,262 ms
34,848 KB
testcase_24 AC 787 ms
34,720 KB
testcase_25 AC 755 ms
34,848 KB
testcase_26 TLE -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define REP(i,n) for(int i=0;i<(int)(n);++i)
#define REPR(i,n) for (int i=(int)(n)-1;i>=0;--i)
#define FOR(i,c) for(__typeof((c).begin())i=(c).begin();i!=(c).end();++i)
#define ALL(c) (c).begin(), (c).end()
#define valid(y,x,h,w) (0<=y&&y<h&&0<=x&&x<w)
#define tpl(...) make_tuple(__VA_ARGS__)
const int INF = 0x3f3f3f3f;
const double EPS = 1e-8;
const double PI = acos(-1);
const int dy[] = {-1,0,1,0};
const int dx[] = {0,1,0,-1};
typedef long long ll;
typedef pair<int,int> pii;
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }
template<typename Ch,typename Tr,typename C,typename=decltype(begin(C()))>basic_ostream<Ch,Tr>& operator<<(basic_ostream<Ch,Tr>&os,
const C& c){os<<'[';for(auto i=begin(c);i!=end(c);++i)os<<(i==begin(c)?"":" ")<<*i;return os<<']';}
template<class S,class T>ostream&operator<<(ostream &o,const pair<S,T>&t){return o<<'('<<t.first<<','<<t.second<<')';}
template<int N,class Tp>void output(ostream&,const Tp&){}
template<int N,class Tp,class,class ...Ts>void output(ostream &o,const Tp&t){if(N)o<<',';o<<get<N>(t);output<N+1,Tp,Ts...>(o,t);}
template<class ...Ts>ostream&operator<<(ostream&o,const tuple<Ts...>&t){o<<'(';output<0,tuple<Ts...>,Ts...>(o,t);return o<<')';}
template<class T>void output(T t,char z=10){if(t<0)t=-t,putchar(45);int c[20];
int k=0;while(t)c[k++]=t%10,t/=10;for(k||(c[k++]=0);k;)putchar(c[--k]^48);putchar(z);}
template<class T>void outputs(T t){output(t);}
template<class S,class ...T>void outputs(S a,T...t){output(a,32);outputs(t...);}
template<class T>void output(T *a,int n){REP(i,n)cout<<a[i]<<(i!=n-1?',':'\n');}
template<class T>void output(T *a,int n,int m){REP(i,n)output(a[i],m);}
template<class T>bool input(T &t){int n=1,c;for(t=0;!isdigit(c=getchar())&&~c&&c-45;);
if(!~c)return 0;for(c-45&&(n=0,t=c^48);isdigit(c=getchar());)t=10*t+c-48;t=n?-t:t;return 1;}
template<class S,class ...T>bool input(S&a,T&...t){input(a);return input(t...);}
template<class T>bool inputs(T *a, int n) { REP(i,n) if(!input(a[i])) return 0; return 1;}

template<int MOD>
struct ModInt {
  int x;
  ModInt() : x(0) {}
  ModInt(ll s) { if ((x = s % MOD) < 0) x += MOD; }
  ModInt operator+=(ModInt rhs) { if ((x+=rhs.x) >= MOD) x -= MOD; return *this; }
  ModInt operator-=(ModInt rhs) { if ((x-=rhs.x) < 0) x += MOD; return *this; }
  ModInt operator*=(ModInt rhs) { x = (ll)x*rhs.x % MOD; return *this; }
  ModInt operator+(ModInt rhs) const { return ModInt(*this) += rhs; }
  ModInt operator-(ModInt rhs) const { return ModInt(*this) -= rhs; }
  ModInt operator*(ModInt rhs) const { return ModInt(*this) *= rhs; }
  ModInt operator/=(ModInt rhs) {
    static const ll inv2 = ModInt(2).inv().x; // 2で割るのは特別に保存してみる
    ll i = (rhs.x == 2 ? inv2 : rhs.inv().x);
    x = x*i%MOD; return *this;
  }
  ModInt operator/(ModInt rhs) const { return ModInt(*this) /= rhs; }
  ModInt inv() { return pow(MOD-2); }
  ModInt pow(ll n) const {
    ModInt r = 1, t = x;
    for (;n;n>>=1,t*=t) if (n&1) r *= t;
    return r;
  }
  static ModInt strnum(const string &n) {ModInt a = 0; for (char c:n) a = a*10+c-'0'; return a;}
  friend ostream& operator<<(ostream &os, const ModInt rhs){ return os << rhs.x; }
};
typedef ModInt<int(1e9+7)> mint;

vector<int> g[2000];
mint dp[2000][2001];
mint dp2[2001][2001];
int num[2000];
void solve(int v, int p) {
  num[v] = 1;
  for (int u : g[v]) {
    if (u == p) continue;
    solve(u,v);
    num[v] += num[u];
  }
  dp2[0][0] = 1;
  int cnt = 0;
  for (int u : g[v]) {
    if (u == p) continue;
    REP(i,num[v]+1) {
      REP(j,num[u]+1) {
        dp2[cnt+1][i+j] += dp2[cnt][i] * dp[u][j];
      }
    }
    cnt++;
  }
  REP(j,num[v]+1) dp[v][j] = dp2[cnt][j];
  dp[v][num[v]] = 1;
  REP(i,cnt+1) REP(j,num[v]+1) dp2[i][j] = 0;
  // cout << "!" << v << " " << dp2[1][0] << endl;
  // REP(i,num[v]+1) {
  //   cout << dp[v][i] << " ";
  // }
  // cout << endl;
}

int main() {
  int n,k;
  while(input(n,k)) {
    REP(i,n) g[i].clear();
    memset(dp,0,sizeof(dp));
    memset(dp2,0,sizeof(dp2));
    REP(i,n-1) {
      int a,b;
      input(a,b);
      g[a].push_back(b);
      g[b].push_back(a);
    }
    solve(0,-1);
    cout << dp[0][k] << endl;
  }
}
0