結果

問題 No.125 悪の花弁
ユーザー PachicobuePachicobue
提出日時 2018-12-21 01:29:12
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 144 ms / 5,000 ms
コード長 1,034 bytes
コンパイル時間 712 ms
コンパイル使用メモリ 81,316 KB
実行使用メモリ 22,868 KB
最終ジャッジ日時 2024-09-25 09:15:14
合計ジャッジ時間 2,029 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 96 ms
17,260 KB
testcase_01 AC 102 ms
14,884 KB
testcase_02 AC 140 ms
19,296 KB
testcase_03 AC 144 ms
19,184 KB
testcase_04 AC 143 ms
22,868 KB
testcase_05 AC 123 ms
19,268 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