結果

問題 No.125 悪の花弁
ユーザー PachicobuePachicobue
提出日時 2018-12-21 01:25:51
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 996 bytes
コンパイル時間 751 ms
コンパイル使用メモリ 80,272 KB
実行使用メモリ 22,784 KB
最終ジャッジ日時 2024-09-25 09:15:09
合計ジャッジ時間 3,115 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 WA -
testcase_05 WA -
権限があれば一括ダウンロードができます

ソースコード

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);
        (sum += pc[p]) %= MOD;
    }
    std::cout << sum * SI % MOD << std::endl;

    return 0;
}
0