結果

問題 No.125 悪の花弁
ユーザー t98slidert98slider
提出日時 2022-06-14 07:01:32
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,194 bytes
コンパイル時間 628 ms
コンパイル使用メモリ 93,080 KB
最終ジャッジ日時 2024-04-27 04:07:45
合計ジャッジ時間 1,103 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:7:5: error: 'cin' was not declared in this scope
    7 |     cin >> K;
      |     ^~~
main.cpp:2:1: note: 'std::cin' is defined in header '<iostream>'; did you forget to '#include <iostream>'?
    1 | #include<atcoder/all>
  +++ |+#include <iostream>
    2 | using namespace std;
main.cpp:40:5: error: 'cout' was not declared in this scope
   40 |     cout << ans.val() << '\n';
      |     ^~~~
main.cpp:40:5: note: 'std::cout' is defined in header '<iostream>'; did you forget to '#include <iostream>'?

ソースコード

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