結果

問題 No.125 悪の花弁
ユーザー PachicobuePachicobue
提出日時 2018-12-21 01:29:12
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 161 ms / 5,000 ms
コード長 1,034 bytes
コンパイル時間 1,058 ms
コンパイル使用メモリ 80,308 KB
実行使用メモリ 22,988 KB
最終ジャッジ日時 2023-10-26 01:15:04
合計ジャッジ時間 2,587 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 104 ms
17,316 KB
testcase_01 AC 114 ms
14,940 KB
testcase_02 AC 161 ms
19,292 KB
testcase_03 AC 159 ms
19,292 KB
testcase_04 AC 156 ms
22,988 KB
testcase_05 AC 138 ms
19,360 KB
権限があれば一括ダウンロードができます

ソースコード

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