結果

問題 No.125 悪の花弁
コンテスト
ユーザー Pachicobue
提出日時 2018-12-21 01:29:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 155 ms / 5,000 ms
コード長 1,034 bytes
コンパイル時間 743 ms
コンパイル使用メモリ 78,264 KB
最終ジャッジ日時 2025-01-06 19:45:52
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <numeric>
using ll = long long;
constexpr ll MOD = 1000000007;
int main()
{
    int K;
    std::cin >> K;
    std::vector<int> C(K);
    int G = -1, S = 0;
    for (int i = 0; i < K; i++) { std::cin >> C[i], G = (G == -1 ? C[i] : std::gcd(G, C[i])), S += C[i]; }
    std::vector<ll> f(S + 1, 1), invf(S + 1, 1);
    for (ll i = 2; i <= (ll)S; i++) { f[i] = (f[i - 1] * i) % MOD, invf[i] = ((MOD - MOD / i) * invf[MOD % i]) % MOD; }
    ll SI = invf[S];
    for (int i = 2; i <= S; i++) { (invf[i] *= invf[i - 1]) %= MOD; }
    std::vector<ll> pc(G + 1, 0);
    for (int p = 1; p <= G; p++) {
        if (G % p != 0) { continue; }
        pc[p] = f[S / p];
        for (int i = 0; i < K; i++) { (pc[p] *= invf[C[i] / p]) %= MOD; }
    }
    ll sum = 0;
    for (int i = 0; i < S; i++) {
        const int p = S / std::gcd(S, i);
        if (G % p != 0) { continue; }
        (sum += pc[p]) %= MOD;
    }
    std::cout << sum * SI % MOD << std::endl;

    return 0;
}
0