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

提出ソース

結果

問題	No.196 典型DP (1)
ユーザー	snuke
提出日時	2015-04-26 22:34:38
言語	C++11 (gcc 11.4.0)
結果	AC
実行時間	54 ms / 2,000 ms
コード長	2,474 bytes
コンパイル時間	708 ms
コンパイル使用メモリ	83,504 KB
実行使用メモリ	40,564 KB
最終ジャッジ日時	2023-09-18 12:05:33
合計ジャッジ時間	3,217 ms
ジャッジサーバーID （参考情報）	judge12 / judge14

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	4 ms 8,460 KB
testcase_01	AC	4 ms 8,536 KB
testcase_02	AC	4 ms 8,372 KB
testcase_03	AC	4 ms 8,436 KB
testcase_04	AC	4 ms 8,388 KB
testcase_05	AC	4 ms 8,456 KB
testcase_06	AC	4 ms 8,356 KB
testcase_07	AC	4 ms 8,360 KB
testcase_08	AC	4 ms 8,396 KB
testcase_09	AC	4 ms 8,392 KB
testcase_10	AC	4 ms 8,388 KB
testcase_11	AC	4 ms 8,392 KB
testcase_12	AC	5 ms 8,392 KB
testcase_13	AC	4 ms 8,500 KB
testcase_14	AC	4 ms 8,488 KB
testcase_15	AC	5 ms 8,540 KB
testcase_16	AC	9 ms 8,792 KB
testcase_17	AC	14 ms 8,840 KB
testcase_18	AC	20 ms 8,948 KB
testcase_19	AC	24 ms 9,120 KB
testcase_20	AC	33 ms 9,180 KB
testcase_21	AC	33 ms 9,132 KB
testcase_22	AC	34 ms 9,100 KB
testcase_23	AC	42 ms 20,580 KB
testcase_24	AC	39 ms 16,888 KB
testcase_25	AC	39 ms 15,724 KB
testcase_26	AC	53 ms 40,564 KB
testcase_27	AC	54 ms 40,384 KB
testcase_28	AC	54 ms 40,396 KB
testcase_29	AC	54 ms 40,312 KB
testcase_30	AC	54 ms 40,304 KB
testcase_31	AC	52 ms 9,012 KB
testcase_32	AC	52 ms 9,228 KB
testcase_33	AC	52 ms 8,944 KB
testcase_34	AC	53 ms 8,992 KB
testcase_35	AC	53 ms 8,968 KB
testcase_36	AC	51 ms 9,224 KB
testcase_37	AC	34 ms 8,980 KB
testcase_38	AC	51 ms 8,960 KB
testcase_39	AC	46 ms 8,928 KB
testcase_40	AC	52 ms 9,184 KB
testcase_41	AC	5 ms 8,396 KB
testcase_42	AC	4 ms 8,384 KB
testcase_43	AC	4 ms 8,356 KB

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

ソースコード

raw source code

#include <cstdio>
#include <algorithm>
#include <stack>
#include <queue>
#include <deque>
#include <vector>
#include <string>
#include <string.h>
#include <cstdlib>
#include <ctime>
#include <cmath>
#include <map>
#include <set>
#include <iostream>
#include <sstream>
#include <cctype>
#define fi first
#define se second
#define rep(i,n) for(int i = 0; i < n; ++i)
#define rrep(i,n) for(int i = 1; i <= n; ++i)
#define drep(i,n) for(int i = n-1; i >= 0; --i)
#define gep(i,g,j) for(int i = g.head[j]; i != -1; i = g.e[i].next)
#define each(it,c) for(__typeof((c).begin()) it=(c).begin();it!=(c).end();it++)
#define rng(a) a.begin(),a.end()
#define maxs(x,y) x = max(x,y)
#define mins(x,y) x = min(x,y)
#define pb push_back
#define sz(x) (int)(x).size()
#define pcnt __builtin_popcount
#define snuke srand((unsigned)clock()+(unsigned)time(NULL));
using namespace std;
typedef long long int ll;
typedef pair<int,int> P;
inline int in() { int x; scanf("%d",&x); return x;}

const int MX = 100005, INF = 1000010000;
const ll LINF = 1000000000000000000ll;
const double eps = 1e-10;
const int di[] = {-1,0,1,0}, dj[] = {0,-1,0,1}; //^<v>

// Mod int
const int mod = 1000000007;
struct mint{
	ll x;
	mint():x(0){}
	mint(ll x):x((x%mod+mod)%mod){}
	mint operator+=(const mint& a){ if((x+=a.x)>=mod) x-=mod; return *this;}
	mint operator-=(const mint& a){ if((x+=mod-a.x)>=mod) x-=mod; return *this;}
	mint operator*=(const mint& a){ (x*=a.x)%=mod; return *this;}
	mint operator+(const mint& a)const{ return mint(*this) += a;}
	mint operator-(const mint& a)const{ return mint(*this) -= a;}
	mint operator*(const mint& a)const{ return mint(*this) *= a;}
  bool operator==(const mint& a)const{ return x == a.x;}
};
//
typedef vector<int> vi;
typedef vector<mint> vm;
typedef vector<vm> vvm;

int n, k;
vi to[MX];

int s[MX];
vvm dp[MX];

void dfs(int v, int p=-1) {
  s[v] = 1;
  dp[v] = vvm(2,vm(2));
  dp[v][0][0] = 1;
  dp[v][1][1] = 1;
  for (int u : to[v]) {
    if (u == p) continue;
    dfs(u,v);
    vvm a = dp[v];
    vvm& b = dp[u];
    rep(i,2) dp[v][i] = vm(s[v]+s[u]+1,0);
    rep(i,s[v]+1)rep(j,s[u]+1) {
      dp[v][0][i+j] += a[0][i]*b[0][j];
      dp[v][0][i+j] += a[0][i]*b[1][j];
      dp[v][1][i+j] += a[1][i]*b[1][j];
    }
    s[v] += s[u];
  }
}

int main(){
  cin >> n >> k;
  rep(i,n-1) {
    int a, b;
    cin >> a >> b;
    to[a].pb(b);
    to[b].pb(a);
  }
  dfs(0);
  mint ans = 0;
  rep(i,2) ans += dp[0][i][k];
  cout << ans.x << endl;
  return 0;
}

yukicoder

結果

テストケース

ソースコード