結果

問題 No.1629 Sorting Integers (SUM of M)
ユーザー yosakayosaka
提出日時 2021-07-31 01:23:18
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,384 bytes
コンパイル時間 861 ms
コンパイル使用メモリ 87,160 KB
実行使用メモリ 126,140 KB
最終ジャッジ日時 2023-10-14 09:37:28
合計ジャッジ時間 3,696 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 126 ms
117,432 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 AC 125 ms
117,448 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import combinations_with_replacement, permutations, combinations

# 逆元のテーブルを作成
def cmb(n, k, mod, fac, ifac):
    k = min(k, n - k)
    return fac[n] * ifac[k] * ifac[n - k] % mod


def make_tables(mod, n):
    fac = [1, 1]  # 階乗テーブル・・・(1)
    ifac = [1, 1]  # 逆元の階乗テーブル・・・(2)
    inverse = [0, 1]  # 逆元テーブル・・・(3)

    for i in range(2, n + 1):
        fac.append((fac[-1] * i) % mod)
        inverse.append((-inverse[mod % i] * (mod // i)) % mod)
        ifac.append((ifac[-1] * inverse[-1]) % mod)
    fac[0] = 0
    ifac[0] = 0
    return fac, ifac


def main():
    N = int(input())
    A = list(map(int, input().split()))
    lst = []
    for i in range(1, 10):
        lst += [i for _ in range(A[i - 1])]
    mod = 10 ** 9 + 7
    fac, ifac = make_tables(mod, 2 * 10 ** 5 + 10)
    ans = 0
    total = fac[N - 1]
    for i in range(1, 10):
        if A[i - 1] == 0:
            continue
        if total > fac[A[i - 1]]:
            total //= fac[A[i - 1]]
        else:
            total += mod
            total //= fac[A[i - 1]]
    # for k in range(N):
    for i in range(1, 10):
        ans += ((pow(10, N, mod) - 1) // 9) * i * A[i - 1]
        ans %= mod
        # print((pow(10, N, mod) - 1) // 9, i, A[i - 1])
    ans *= max(1, total)
    ans %= mod
    print(ans)


main()
0