結果

問題 No.1455 拡張ROTN
コンテスト
ユーザー Theta
提出日時 2022-10-13 18:59:33
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 36 ms / 2,000 ms
コード長 899 bytes
コンパイル時間 364 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-06-26 12:07:45
合計ジャッジ時間 1,736 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import Counter


def extended_rotate(letter: str) -> str:
    if letter == "z":
        return "a"
    if letter == "Z":
        return "A"
    if letter == "9":
        return "CpCzNkSuTbEoA"

    return chr(ord(letter) + 1)


def main():
    S = input()
    N = int(input())

    while N > 0:
        counter_S = Counter(S)
        if sum(counter_S[str(i)] for i in range(10)):
            S_tmp = ""
            for letter in S:
                S_tmp += extended_rotate(letter)
            S = S_tmp
            N -= 1
        else:
            N %= 26
            break

    while N > 0:
        counter_S = Counter(S)
        if sum(counter_S[str(i)] for i in range(10)):
            raise ValueError
        S_tmp = ""
        for letter in S:
            S_tmp += extended_rotate(letter)
        S = S_tmp
        N -= 1

    print(S)


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