結果

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

ソースコード

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) {
    cerr << x << ' ' << prev << endl;
    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;
    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