結果

問題 No.125 悪の花弁
ユーザー CaiiiiiiiiCaiiiiiiii
提出日時 2024-03-08 16:44:33
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 23 ms / 5,000 ms
コード長 1,135 bytes
コンパイル時間 2,468 ms
コンパイル使用メモリ 202,712 KB
実行使用メモリ 11,992 KB
最終ジャッジ日時 2024-09-29 18:48:53
合計ジャッジ時間 2,530 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 11 ms
11,848 KB
testcase_01 AC 19 ms
10,656 KB
testcase_02 AC 22 ms
11,768 KB
testcase_03 AC 23 ms
11,992 KB
testcase_04 AC 14 ms
11,420 KB
testcase_05 AC 13 ms
11,408 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

const int N = 1e5 + 7;
const int M = 1e6 + 7;
const int MOD = 1e9 + 7;

int fac[M], ifc[M], n, a[N], f[N], d[N], cd;

int pow(int x, int y) {
  int ret = 1;
  while(y) {
    if(y & 1)
      ret = 1LL * ret * x % MOD;
    x = 1LL * x * x % MOD;
    y >>= 1;
  }
  return ret;
}

int main() {

  scanf("%d", &n);
  for(int i = 1; i <= n; ++i)
    scanf("%d", &a[i]);

  int sum = 0, g = 0;
  for(int i = 1; i <= n; ++i) {
    sum += a[i];
    g = std::__gcd(g, a[i]);
  }

  fac[0] = 1;
  for(int i = 1; i <= sum; ++i)
    fac[i] = 1LL * fac[i - 1] * i % MOD;
  ifc[sum] = pow(fac[sum], MOD - 2);
  for(int i = sum - 1; i >= 0; --i)
    ifc[i] = 1LL * ifc[i + 1] * (i + 1) % MOD;

  for(int i = 1; i <= g; ++i)
    if(g % i == 0)
      d[++cd] = i;

  int ans = 0;
  for(int i = cd; i >= 1; --i) {
    f[i] = fac[sum / d[i]];
    for(int j = 1; j <= n; ++j)
      f[i] = 1LL * f[i] * ifc[a[j] / d[i]] % MOD;
    for(int j = cd; j > i; --j)
      if(d[j] % d[i] == 0)
	f[i] = (f[i] + MOD - f[j]) % MOD;
    ans = (ans + 1LL * f[i] * pow(sum / d[i], MOD - 2)) % MOD;
  }

  printf("%d\n", ans);
  
  return 0;

}
0