結果

問題 No.758 VVVVV
ユーザー nu50218
提出日時 2019-09-03 00:24:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 80 ms / 2,000 ms
コード長 1,467 bytes
コンパイル時間 1,677 ms
コンパイル使用メモリ 175,160 KB
実行使用メモリ 9,980 KB
最終ジャッジ日時 2024-10-14 02:00:22
合計ジャッジ時間 3,420 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

const int MOD = 1000000007;

long long pow(long long x, long long n, long long mod) {
    long long ans = 1;
    while (n) {
        if (n % 2) ans *= x;
        x *= x;
        x %= mod;
        ans %= mod;
        n >>= 1;
    }
    return ans;
}
vector<long long> factorial(2, 1);
long long comb(long long n, long long k, long long p) {
    if (n < k || n <= 0 || k < 0) return 0;
    while (factorial.size() <= n) {
        long long i = factorial.size();
        factorial.push_back(i * factorial[i - 1] % p);
    }
    return ((factorial[n] * pow(factorial[k], p - 2, p)) % p) *
           pow(factorial[n - k], p - 2, p) % p;
}

long long Narayana(const long long& n, const long long& k, const long long& p) {
    return comb(n, k, p) * comb(n, k - 1, p) % p * pow(n, MOD - 2, p) % p;
}

vector<vector<int>> adj;
int leaf_count = 0;

void dfs(int x, int prev) {
    bool is_leaf = true;
    for (auto next : adj[x]) {
        if (next != prev) {
            dfs(next, x);
            is_leaf = false;
        }
    }
    leaf_count += is_leaf;
}

int main() {
    int N;
    cin >> N;
    if (N == 1) {
        cout << 1 << endl;
        return 0;
    }
    adj.resize(N);
    for (int i = 0; i < N - 1; i++) {
        int a, b;
        cin >> a >> b;
        a--;
        b--;
        adj[a].push_back(b);
        adj[b].push_back(a);
    }
    dfs(0, -1);
    cout << Narayana(N - 1, leaf_count, MOD) << endl;
}
0