結果

問題 No.196 典型DP (1)
ユーザー maine_honzuki
提出日時 2020-12-16 12:36:32
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 20 ms / 2,000 ms
コード長 882 bytes
コンパイル時間 2,112 ms
コンパイル使用メモリ 203,276 KB
最終ジャッジ日時 2025-01-17 01:54:13
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const ll MOD = 1e9 + 7;

vector<vector<int>> G;

void dfs(int idx, int par, vector<ll>& V) {
    V.emplace_back(1);
    for (auto& nxt : G[idx]) {
        if (nxt == par)
            continue;
        int N = V.size();
        vector<ll> V_nxt, V_memo = V;
        dfs(nxt, idx, V_nxt);
        V.resize(V.size() + V_nxt.size() - 1);
        for (int i = 1; i < V_nxt.size(); i++) {
            for (int j = 0; j < N; j++) {
                (V[j + i] += V_memo[j] * V_nxt[i]) %= MOD;
            }
        }
    }
    V.emplace_back(1);
}

int main() {
    int N, K;
    cin >> N >> K;
    G.resize(N);
    for (int i = 0; i < N - 1; i++) {
        int a, b;
        cin >> a >> b;
        G[a].emplace_back(b);
        G[b].emplace_back(a);
    }

    vector<ll> V;
    dfs(0, -1, V);
    cout << V[K] << endl;
}
0