結果

問題 No.1595 The Final Digit
コンテスト
ユーザー ygd.
提出日時 2021-07-31 17:09:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 47 ms / 2,000 ms
コード長 787 bytes
コンパイル時間 163 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 54,016 KB
最終ジャッジ日時 2024-09-16 09:34:17
合計ジャッジ時間 1,801 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict 

def main():
    p,q,r,k = map(int,input().split())
    A = [p%10,q%10,r%10]
    dic = defaultdict(int)
    dic[tuple(A)] = 1 #1スタート
    dic1 = defaultdict(int)
    num = 2
    while True:
        val = A[-1] + A[-2] + A[-3]
        val %= 10
        nxt = (A[-2], A[-1], val)
        A.append(val)
        if nxt in dic:
            loop_start = dic[nxt]
            loop_last = num - 1 
            break
        else:
            dic[nxt] = num
            num += 1
    #print(dic,loop_start,loop_last)
    loop_len = loop_last - loop_start + 1 
    if k <= len(A):
        print(A[k-1]);exit()
    amari = loop_start - 1
    idx = (k - amari) % loop_len + amari - 1
    #print(idx)
    print(A[idx])

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