結果

問題 No.434 占い
コンテスト
ユーザー rpy3cpp
提出日時 2016-10-18 23:13:07
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 152 ms / 2,000 ms
コード長 656 bytes
コンパイル時間 181 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-11-22 12:58:02
合計ジャッジ時間 2,925 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

inv = [0, 1, 5, 0, 7, 2, 0, 4, 8, 0]

def solve(S, inv):
    pow3 = 0
    cum = 1
    ans = int(S[0])
    lenS = len(S)
    if lenS == 1:
        return ans
    if S == '0' * lenS:
        return 0
    num = lenS - 1
    for i in range(1, lenS):
        n = num
        d = lenS - n
        while not n % 3:
            pow3 += 1
            n //= 3
        while not d % 3:
            pow3 -= 1
            d //= 3
        cum = (cum * n * inv[d % 9]) % 9
        if pow3 < 2:
            ans += cum * int(S[i]) * 3 ** pow3
        num -= 1
    return (ans - 1) % 9 + 1

T = int(input())
for S in sys.stdin:
    print(solve(S.rstrip(), inv))
0