#168135 (C++14) No.196 典型DP (1)

提出ソース

結果

問題	No.196 典型DP (1)
ユーザー	はまやんはまやん
提出日時	2017-04-23 15:03:50
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	401 ms / 2,000 ms
コード長	1,601 bytes
コンパイル時間	1,849 ms
コンパイル使用メモリ	181,680 KB
実行使用メモリ	98,048 KB
最終ジャッジ日時	2024-07-21 20:49:30
合計ジャッジ時間	10,390 ms
ジャッジサーバーID （参考情報）	judge3 / judge1

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	2 ms 5,376 KB
testcase_12	AC	2 ms 5,376 KB
testcase_13	AC	3 ms 5,376 KB
testcase_14	AC	3 ms 5,376 KB
testcase_15	AC	9 ms 5,376 KB
testcase_16	AC	30 ms 5,376 KB
testcase_17	AC	71 ms 5,376 KB
testcase_18	AC	139 ms 5,376 KB
testcase_19	AC	174 ms 5,504 KB
testcase_20	AC	274 ms 5,376 KB
testcase_21	AC	269 ms 5,376 KB
testcase_22	AC	266 ms 5,504 KB
testcase_23	AC	335 ms 38,144 KB
testcase_24	AC	308 ms 27,392 KB
testcase_25	AC	302 ms 24,448 KB
testcase_26	AC	400 ms 97,920 KB
testcase_27	AC	401 ms 98,048 KB
testcase_28	AC	401 ms 97,920 KB
testcase_29	AC	396 ms 98,048 KB
testcase_30	AC	394 ms 97,920 KB
testcase_31	AC	334 ms 5,376 KB
testcase_32	AC	339 ms 5,376 KB
testcase_33	AC	343 ms 5,376 KB
testcase_34	AC	337 ms 5,376 KB
testcase_35	AC	336 ms 5,376 KB
testcase_36	AC	315 ms 5,376 KB
testcase_37	AC	186 ms 5,376 KB
testcase_38	AC	318 ms 5,376 KB
testcase_39	AC	278 ms 5,376 KB
testcase_40	AC	327 ms 5,376 KB
testcase_41	AC	2 ms 5,376 KB
testcase_42	AC	2 ms 5,376 KB
testcase_43	AC	2 ms 5,376 KB

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

ソースコード

raw source code

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a;i<b;i++)




int mod = 1000000007;
int add(int x, int y) { return (x += y) >= mod ? x - mod : x; }
template<class... T> int add(int x, T... y) { return add(x, add(y...)); }
int mul(int x, int y) { return 1LL * x * y % mod; }
template<class... T> int mul(int x, T... y) { return mul(x, mul(y...)); }
int sub(int x, int y) { return add(x, mod - y); }
int modpow(int a, long long b) { int ret = 1; while (b > 0) { if (b & 1) ret = 1LL * ret * a % mod; a = 1LL * a * a % mod; b >>= 1; } return ret; }
int modinv(int a) { return modpow(a, mod - 2); }
//-----------------------------------------------------------------------------------
int N, K;
vector<int> E[2010];
//-----------------------------------------------------------------------------------
int cnt[2010]; map<int,int> dp[2010];
void merge(int cu, int to) { // dp[cu] = dp[cu] + dp[to]
    map<int, int> cnt;
    for (auto p : dp[cu]) for (auto q : dp[to]) {
        cnt[p.first + q.first] = add(cnt[p.first + q.first], mul(p.second, q.second));
    }
    dp[cu] = cnt;
}
void dfs(int cu, int pa = -1) {
    dp[cu][0] = 1;

    cnt[cu] = 1;
    for (int to : E[cu]) if (pa != to) {
        dfs(to, cu);
        cnt[cu] += cnt[to];

        merge(cu, to);
    }

    dp[cu][cnt[cu]] = 1;
}
//-----------------------------------------------------------------------------------
int main() {
    cin >> N >> K;
    rep(i, 0, N - 1) {
        int a, b; scanf("%d%d", &a, &b);
        E[a].push_back(b); E[b].push_back(a);
    }

    dfs(0);
    printf("%d\n", dp[0][K]);
}

yukicoder

結果

テストケース

ソースコード