LeetCode 796. 旋转字符串（两种解法）

LeetCode 796. 旋转字符串

题目

给定两个字符串, A 和 B。

A 的旋转操作就是将 A 最左边的字符移动到最右边。 例如, 若 A = ‘abcde’，在移动一次之后结果就是’bcdea’ 。如果在若干次旋转操作之后，A 能变成B，那么返回True。

示例 1:

输入: A = 'abcde', B = 'cdeab'
输出: true

示例 2:

输入: A = 'abcde', B = 'abced'
输出: false

两种思路

穷举法

这道题穷举也可以通过，只要把A旋转可以得到的所有的情况都穷举出来，跟B做比较就可以了。

class Solution:
    def rotateString(self, A: str, B: str) -> bool:
        if A == B:
            return True
        for _ in range(len(A)):
            A = A + A[0]
            A = A[1:]
            
            if A == B:
                return True
        return False

更简洁的做法

根据题目意思可以知道，如果B包含在A+A当中，那么A总是可以通过某几次旋转得到B。在这里要注意先判断A和B的长度，如果长度不等，那么直接返回False。如果不先进行长度的判断，可能会出现A=‘aa’，B=‘a’而返回True的情况。

class Solution:
    def rotateString(self, A: str, B: str) -> bool:
        if len(A) != len(B):
            return False
        return B in A + A