Pythonの文字列スライスで文字列を回転させる方法

文字列が与えられたとき、その文字列を2つの方向にスライスして回転させるのが本記事の目的です。1つは時計回り(右回転)、もう1つは反時計回り(左回転)です。

1. 与えられた文字列を d 要素分だけ左(反時計回り)に回転させます(d <= n)。

2. 与えられた文字列を d 要素分だけ右(時計回り)に回転させます(d <= n)。

実行例

Input: string = "pythonprogram"
d = 2
Output:
Left Rotation: thonprogrampy
Right Rotation: ampythonprogr

アルゴリズム

Step 1: 文字列を入力する。
Step 2: 文字列を2つの部分に分割する。
        左回転の場合:Lfirst = str[0 : d]、Lsecond = str[d :]
        右回転の場合:Rfirst = str[0 : len(str)-d]、Rsecond = str[len(str)-d : ]
Step 3: 分割した2つの部分を「後半 + 前半」の順に連結する。

仕組みのポイント

Pythonのスライス記法 str[a:b] を使うと、インデックス a から b-1 までの部分文字列を簡単に取り出せます。この性質を利用して、文字列を「先頭からd番目まで」と「d番目以降」の2つに分割し、順序を入れ替えて連結することで回転を実現しています。

• 左回転:先頭の d 文字を末尾へ移動させる(Lsecond + Lfirst)
• 右回転:末尾の d 文字を先頭へ移動させる(Rsecond + Rfirst)

サンプルコード

def rotate(input, d):
    # 文字列を左右回転用に2つの部分へスライス
    Lfirst = input[0 : d]
    Lsecond = input[d :]
    Rfirst = input[0 : len(input)-d]
    Rsecond = input[len(input)-d : ]
    print("Left Rotation : ", (Lsecond + Lfirst))
    print("Right Rotation : ", (Rsecond + Rfirst))

# ドライバープログラム
if __name__ == "__main__":
    s = input("Enter String ::>")
    d = 2
    rotate(s, d)

出力結果

Enter String ::> pythonprogram
Left Rotation: thonprogrampy
Right Rotation: ampythonprogr

このように、Pythonのスライスと文字列連結を組み合わせるだけで、ループ処理を書かずに簡潔に文字列の回転を実装できます。ぜひ実際にコードを動かして挙動を確認してみてください。