結果

問題 No.1455 拡張ROTN
コンテスト
ユーザー kept1994
提出日時 2021-09-20 12:54:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 78 ms / 2,000 ms
コード長 4,858 bytes
コンパイル時間 158 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 70,784 KB
最終ジャッジ日時 2024-07-02 13:37:53
合計ジャッジ時間 2,512 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
import sys

def main():
    def smallConvert(s: str):
        return ord(s) - ord("a")
    def largeConvert(s: str):
        return ord(s) - ord("A") + 26
    
    def decode(i: int):
        if i < 26:
            return chr(i + ord("a"))
        else:
            return chr(i + ord("A") - 26)

        
    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):
            l = []
            for i in range(smallConvert("a"), smallConvert("z")):
                l.append(i + 1)
            l.append(smallConvert("a")) # z
            for i in range(largeConvert("A"), largeConvert("Z")):
                l.append(i + 1)
            l.append(largeConvert("A")) # Z
            self.dv.append(l)
            return

        # ダブリング実施(和を含まない)
        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]

    S = list(input())
    N = int(input())
    import math
    d = Doubling(stateKind=52, maxDoublingTimes=int(math.log2(N)) + 1, useSum=False)
    ans = []
    for ss in S:
        if ss.isdigit():
            rest = int(ss) + N - 9
            if rest <= 0:
                ans.append(str(int(ss) + N))
            else:
                for ch in "CpCzNkSuTbEoA":
                    sss = smallConvert(ch) if ch.islower() else largeConvert(ch)
                    ans.append(decode(d.getState(doubingTimes=rest - 1, startState=sss)))
        else:
            sss = smallConvert(ss) if ss.islower() else largeConvert(ss)
            ans.append(decode(d.getState(doubingTimes=N, startState=sss)))
    print(*ans, sep="")
    return
        
if __name__ == '__main__':
    main()
0