結果

問題 No.1595 The Final Digit
ユーザー kept1994
提出日時 2021-08-27 03:15:08
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 56 ms / 2,000 ms
コード長 4,113 bytes
コンパイル時間 373 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 65,536 KB
最終ジャッジ日時 2024-11-19 03:17:13
合計ジャッジ時間 2,176 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
import sys


def main():
    def converter(i):
        S = str(i)
        if len(S) == 1:
            return [0, 0, i]
        elif len(S) == 2:
            return [0, int(S[0]), int(S[1])]
        else:
            return [int(S[0]), int(S[1]), int(S[2])]

    class Doubling:
        """ 初期化処理。
        Parameters:
        ----------
        stateKind : int
            状態の数。dvテーブルの横の長さ。
        maxDoublingTimes : int
            ダブリングの回数。dvテーブルの縦の長さ。
        useSum : bool
            和のダブリングを使用するかの選択。
        Returns:
        ----------
        None
        """
        def __init__(self, stateKind: int, maxDoublingTimes: int, useSum: bool = False):
            self.dv = []                                # 数列(状態)のダブリングテーブル。dv[k][s] := 状態sを2^k回実行したらあとの状態
            self.sum = []                               # 和のダブリングテーブル
            self.stateKind = stateKind                  # 状態の種類数s
            self.maxDoublingTimes = maxDoublingTimes    # 実行回数kの範囲の定義(2^0 ≦ k ≦ 2^maxDoublingTimes)
            # --- Initialize -------------------
            # STEP.1 テーブルの初期化 2^0(=1)回操作後の状態を生成。
            self._initTable()
            # STEP.2 テーブルの更新。
            if useSum:
                self._createTableWithSum()
            else:
                self._createTable()
            # ---------------------------------

        # 初期化処理
        # 初期化処理は問題毎に記述する。
        def _initTable(self) -> None:
            ll = []
            for i in range(1000):
                l = converter(i)
                ll.append(l[1] * 100 + l[2] * 10 + sum(l) % 10)
            self.dv.append(ll)
        

        # ダブリング実施(和を含まない)
        def _createTable(self):
            for i in range(1, self.maxDoublingTimes):
                l = []
                for j in range(self.stateKind):
                    l.append(self.dv[i - 1][self.dv[i - 1][j]])
                self.dv.append(l)

        # ダブリング実施(和を含む)
        def _createTableWithSum(self):
            for i in range(1, self.maxDoublingTimes):
                l = []
                s = []
                for j in range(self.stateKind):
                    l.append(self.dv[i - 1][self.dv[i - 1][j]])
                    s.append(self.sum[i - 1][j] + self.sum[i - 1][self.dv[i - 1][j]])
                self.dv.append(l)
                self.sum.append(s)
        
        """ 指定回数操作後の状態を算出する。
        Parameters:
        ----------
        doubingTimes : int
            求める状態に至る操作回数。
        startState : int
            開始する状態。
        Returns:
        ----------
        int
            求めるべく状態
        """
        def getState(self, doubingTimes: int, startState: int):
            a = []
            for i in range(self.maxDoublingTimes):
                if doubingTimes >> i & 1:
                    a.append(i)
            now = startState
            for i in a:
                now = self.dv[i][now]
            return now
        
        """
        """
        def getSum(self, doubingTimes: int, startState: int):
            res = 0
            a = []
            for i in range(self.maxDoublingTimes):
                if doubingTimes >> i & 1:
                    a.append(i)
            now = startState
            for i in a:
                res += self.sum[i][now]
                now = self.dv[i][now]
            return res

        def getAllStates(self, targenTime: int):
            return self.dv[targenTime]

    p, q, r, K = map(int, input().split())
    num = (p % 10) * 100 + (q % 10) * 10 + (r % 10)
    d = Doubling(1000, 60)
    print(str(d.getState(startState=num, doubingTimes=(K - 3)))[-1])


    return

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