結果

問題 No.434 占い
コンテスト
ユーザー titia
提出日時 2023-04-25 06:36:14
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,084 ms / 2,000 ms
コード長 1,239 bytes
コンパイル時間 309 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 18,688 KB
最終ジャッジ日時 2024-11-14 03:21:27
合計ジャッジ時間 16,811 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

# エラトステネスの篩を用いた素因数分解・約数列挙
MAX=2*10**5+10 # 使いたい最大値を指定

# Sieve[i]で、iの最も小さい約数を返す。
Sieve=[i for i in range(MAX)]

for i in range(2,MAX):
    if Sieve[i]!=i:
        continue
    
    for j in range(i,MAX,i):
        if Sieve[j]==j:
            Sieve[j]=i

# 素因数分解
def fact(x):
    D=dict()
    while x!=1:
        k=Sieve[x]
        if k in D:
            D[k]+=1
        else:
            D[k]=1
        x//=k
    return D

T=int(input())
for tests in range(T):
    S=list(map(int,list(input())))

    if S==[0]*len(S):
        print(0)
        continue

    LIST=[0]*9

    ANS=0

    for i in range(len(S)):
        if i==0 or i==len(S)-1:
            ANS=(ANS+S[i])%9
            continue

        D=fact(i)
        #print(D)

        for x in D:
            LIST[x%9]-=D[x]

        D=fact(len(S)-i)

        #print(D)

        for x in D:
            LIST[x%9]+=D[x]

        if LIST[0]>0:
            continue

        #print(LIST)
        X=1
        for j in range(9):
            X=X*pow(j,LIST[j],9)%9

        ANS=(ANS+S[i]*X)%9

    if ANS==0:
        ANS=9

    print(ANS)
0