結果

問題 No.1632 Sorting Integers (GCD of M)
ユーザー tamato
提出日時 2021-07-31 11:23:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 41 ms / 2,000 ms
コード長 1,722 bytes
コンパイル時間 250 ms
コンパイル使用メモリ 82,828 KB
実行使用メモリ 54,424 KB
最終ジャッジ日時 2024-09-21 17:33:13
合計ジャッジ時間 4,262 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 59
権限があれば一括ダウンロードができます

ソースコード

diff #

mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    N = int(input())
    C = [0] + list(map(int, input().split()))

    nz = [0] * 10
    for i in range(1, 10):
        if C[i]:
            nz[i] = 1
    if sum(nz) == 1:
        c = 0
        for i in range(1, 10):
            if nz[i]:
                c = i
                break
        print((c * (pow(10, N, mod) - 1) * pow(9, mod-2, mod))%mod)
        exit()

    ans = 1
    # 2
    flg2 = 1
    for i in range(1, 10):
        if i % 2 and C[i]:
            flg2 = 0
            break
    if flg2:
        ans = 2

    # 4
    flg4 = 1
    for i in range(1, 10):
        if i % 4 and C[i]:
            flg4 = 0
            break
    if flg4:
        ans = 4

    # 3, 9
    S = 0
    for i, c in enumerate(C):
        S = S + c * i
    if S % 9 == 0:
        ans *= 9
    elif S % 3 == 0:
        ans *= 3

    # 7
    T = set()
    for i in range(1, 10):
        if C[i]:
            T.add(i)
    T = sorted(list(T))
    if T == [1, 8] or T == [2, 9]:
        if N % 6 == 0:
            ans *= 7

    # 27
    mod27 = [0] * 27
    for i in range(27):
        mod27[i] = (mod27[i-1] + pow(10, i, 27)) % 27

    U = set()
    for i in range(1, 10):
        if C[i]:
            U.add(i)
    U = sorted(list(U))
    flg27 = 1
    for u1 in U:
        for u2 in U:
            if (u1 - u2) % 3:
                flg27 = 0
    if flg27:
        cnt = 0
        V = 0
        for i, c in enumerate(C):
            if c:
                V += i * (mod27[(c + cnt - 1) % 27] - mod27[(cnt - 1) % 27])
                cnt += c % 27
        if V % 27 == 0:
            ans *= 3

    print(ans)


if __name__ == '__main__':
    main()
0