#22736 (C++11) No.196 典型DP (1)

提出ソース

結果

問題	No.196 典型DP (1)
ユーザー	sune232002
提出日時	2015-04-27 01:18:14
言語	C++11 (gcc 11.4.0)
結果	AC
実行時間	20 ms / 2,000 ms
コード長	4,210 bytes
コンパイル時間	1,315 ms
コンパイル使用メモリ	162,280 KB
実行使用メモリ	19,432 KB
最終ジャッジ日時	2024-07-05 03:27:58
合計ジャッジ時間	3,202 ms
ジャッジサーバーID （参考情報）	judge5 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	9 ms 19,032 KB
testcase_01	AC	10 ms 19,072 KB
testcase_02	AC	9 ms 19,024 KB
testcase_03	AC	9 ms 19,072 KB
testcase_04	AC	8 ms 18,996 KB
testcase_05	AC	8 ms 18,944 KB
testcase_06	AC	8 ms 19,072 KB
testcase_07	AC	8 ms 19,012 KB
testcase_08	AC	8 ms 19,200 KB
testcase_09	AC	9 ms 19,160 KB
testcase_10	AC	7 ms 18,984 KB
testcase_11	AC	9 ms 19,128 KB
testcase_12	AC	10 ms 19,156 KB
testcase_13	AC	9 ms 19,072 KB
testcase_14	AC	8 ms 19,192 KB
testcase_15	AC	9 ms 19,072 KB
testcase_16	AC	10 ms 19,024 KB
testcase_17	AC	10 ms 19,200 KB
testcase_18	AC	12 ms 18,944 KB
testcase_19	AC	11 ms 19,072 KB
testcase_20	AC	13 ms 19,072 KB
testcase_21	AC	13 ms 19,140 KB
testcase_22	AC	14 ms 19,200 KB
testcase_23	AC	15 ms 19,072 KB
testcase_24	AC	14 ms 19,072 KB
testcase_25	AC	14 ms 19,120 KB
testcase_26	AC	20 ms 19,328 KB
testcase_27	AC	20 ms 19,236 KB
testcase_28	AC	18 ms 19,432 KB
testcase_29	AC	18 ms 19,360 KB
testcase_30	AC	18 ms 19,200 KB
testcase_31	AC	17 ms 19,200 KB
testcase_32	AC	17 ms 19,200 KB
testcase_33	AC	14 ms 19,120 KB
testcase_34	AC	16 ms 18,944 KB
testcase_35	AC	15 ms 19,200 KB
testcase_36	AC	15 ms 19,072 KB
testcase_37	AC	13 ms 19,072 KB
testcase_38	AC	15 ms 19,072 KB
testcase_39	AC	14 ms 19,200 KB
testcase_40	AC	15 ms 19,072 KB
testcase_41	AC	9 ms 19,184 KB
testcase_42	AC	8 ms 19,188 KB
testcase_43	AC	8 ms 19,008 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#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];
int num[2000];
void solve(int v, int p) {
  num[v] = 1;
  for (int u : g[v]) {
    if (u == p) continue;
    solve(u,v);
  }
  int cnt = 0;
  dp[v][0] = 1;
  for (int u : g[v]) {
    if (u == p) continue;
    for (int i=num[v]; i>=0; --i) {
      for (int j=num[u]; j>=1; --j) {
        dp[v][i+j] += dp[v][i] * dp[u][j];
      }
    }
    num[v] += num[u];
    cnt++;
  }
  dp[v][num[v]] = 1;
  // 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));
    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;
  }
}

yukicoder

結果

テストケース

ソースコード