結果

問題 No.1629 Sorting Integers (SUM of M)
コンテスト
ユーザー Nishin-0141
提出日時 2022-01-28 19:46:35
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,502 ms / 2,000 ms
コード長 737 bytes
コンパイル時間 96 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 26,408 KB
最終ジャッジ日時 2024-12-29 23:21:10
合計ジャッジ時間 10,781 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

def per(n: int, r: list, mod: int):
    """同じものを含む順列の計算"""
    global fact, factinv
    num = fact[n]
    for i in r:
        if i:
            num *= factinv[i]
            num %= mod
    return num


n = int(input())
c = list(map(int, input().split()))
mod = 10**9 + 7

# n までのそれぞれの階乗計算（mod 含む）
fact = [1, 1]
factinv = [1, 1]
for i in range(2, n+1):
    fact.append(fact[-1] * i % mod)
    factinv.append(pow(fact[-1], mod - 2, mod))

cnt = 0
for i in range(9):
    check = False
    if c[i]:
        c[i] -= 1
        cnt += per(n - 1, c, mod) * (i + 1) % mod
        c[i] += 1
# print(cnt)

ans = 0
for i in range(n):
    ans += cnt * pow(10, i, mod)
    ans %= mod
print(ans)
0