結果

問題 No.125 悪の花弁
ユーザー shotoyooshotoyoo
提出日時 2021-06-24 00:16:06
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 334 ms / 5,000 ms
コード長 1,026 bytes
コンパイル時間 554 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 162,204 KB
最終ジャッジ日時 2024-06-25 00:26:26
合計ジャッジ時間 2,658 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 245 ms
125,056 KB
testcase_01 AC 239 ms
135,424 KB
testcase_02 AC 311 ms
158,336 KB
testcase_03 AC 334 ms
162,204 KB
testcase_04 AC 333 ms
152,032 KB
testcase_05 AC 303 ms
140,628 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

sys.setrecursionlimit(2*10**5+10)
write = lambda x: sys.stdout.write(x+"\n")
debug = lambda x: sys.stderr.write(x+"\n")
writef = lambda x: print("{:.12f}".format(x))

### 約数列挙
def fs(n):
    s = set()
    for i in range(1,int(n**0.5)+2):
        if n%i==0:
            s.add(i)
            s.add(n//i)
    l = sorted(list(s))
    return l

k = int(input())
c = list(map(int, input().split()))
M = 10**9+7
ans = 0
size = 0
n = sum(c)
from math import gcd
g = c[0]
for i in range(1,k):
    g = gcd(g, c[i])
g1 = [1]
for i in range(1, n+1):
    g1.append(g1[-1]*i%M)
f = fs(g)
_inv = {}
def inv(v):
    if v in _inv:
        return _inv[v]
    val = pow(v, M-2, M)
    _inv[v] = val
    return val
for v in range(1,n+1):
    tmp = gcd(v,n)
    v = tmp
    d = n//v
    if g%d!=0:
        continue
    val = g1[v]
    for cc in c:
        val *= inv(g1[cc//d])
        val %= M
    ans += val
    ans %= M
#     print(v, d, val)
ans *= pow(n, M-2, M)
print(ans%M)
0