結果

問題 No.125 悪の花弁
ユーザー t98slidert98slider
提出日時 2022-06-14 07:01:32
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 37 ms / 5,000 ms
コード長 1,194 bytes
コンパイル時間 3,383 ms
コンパイル使用メモリ 144,100 KB
実行使用メモリ 11,340 KB
最終ジャッジ日時 2023-07-25 04:20:11
合計ジャッジ時間 4,216 ms
ジャッジサーバーID
(参考情報)
judge11 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 12 ms
8,780 KB
testcase_01 AC 28 ms
9,076 KB
testcase_02 AC 32 ms
11,340 KB
testcase_03 AC 37 ms
11,136 KB
testcase_04 AC 16 ms
10,864 KB
testcase_05 AC 16 ms
10,856 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<atcoder/all>
using namespace std;
using mint = atcoder::modint1000000007;

int main(){
    int K, Csum = 0, gcdv = 0;
    cin >> K;
    vector<int> C(K), y;
    for(auto &&v:C){
        cin >> v;
        gcdv = __gcd(gcdv, v);
        Csum += v;
    }
    vector<mint> fact(Csum + 1), inv(Csum + 1);
    fact[0] = 1;
    for(int i = 1; i <= Csum; i++)fact[i] = i * fact[i - 1];
    inv[Csum] = 1 / fact[Csum];
    for(int i = Csum; i >= 1; i--)inv[i - 1] = i * inv[i];
    auto comb = [&](int n, int k){ return fact[n] * inv[n - k] * inv[k]; };
    for(int i = 1; i * i <= gcdv; i++){
        if(gcdv % i == 0){
            y.push_back(i);
            if(gcdv / i != i)y.push_back(gcdv / i);
        }
    }
    sort(y.rbegin(), y.rend());
    vector<mint> calc(y.size(), 1);
    mint ans;
    for(int i = 0; i < y.size(); i++){
        int d = y[i], rem = Csum / d - 1;
        for(int j = 0; j < K; j++){
            calc[i] *= comb(rem, C[j] / d - (j == 0));
            rem -= C[j] / d - (j == 0);
        }
        for(int j = 0; j < i; j++){
            if(y[j] % y[i] == 0)calc[i] -= calc[j];
        }
        ans += calc[i] / (C[0] / d);
    }
    cout << ans.val() << '\n';
}
0