旋转字符串：解法与代码实现

2024-02-23 15:27:09

旋转字符串是一个有趣且具有挑战性的问题。它要求我们能够模拟字符串的旋转操作，并判断两个字符串是否相等。这篇文章将指导你完成这个问题的解决步骤，并提供详细的代码实现。

解题思路

这道题的本质是判断两个字符串是否相等。我们可以通过模拟字符串的旋转操作来判断它们是否相等。具体步骤如下：

将第一个字符串 s 旋转 i 次，得到一个新的字符串 s_i。
将 s_i 与第二个字符串 t 进行比较。如果 s_i 等于 t，则返回 true。
如果 s_i 不等于 t，则将 i 加 1，并重复步骤 1 和 2。
如果 i 等于字符串 s 的长度，则返回 false。

代码实现

def rotate_string(s, i):
  """
  旋转字符串 s i 次。

  参数：
    s: 要旋转的字符串
    i: 旋转的次数

  返回：
    旋转后的字符串
  """

  # 将字符串 s 转换为列表
  s_list = list(s)

  # 将字符串 s 旋转 i 次
  for _ in range(i):
    # 将字符串 s 的最后一个字符移到第一个字符的位置
    last_char = s_list.pop()
    s_list.insert(0, last_char)

  # 将字符串 s_list 转换为字符串并返回
  return ''.join(s_list)


def is_equal(s, t):
  """
  判断两个字符串是否相等。

  参数：
    s: 第一个字符串
    t: 第二个字符串

  返回：
    如果两个字符串相等，则返回 true，否则返回 false
  """

  # 如果两个字符串的长度不相等，则返回 false
  if len(s) != len(t):
    return False

  # 循环比较两个字符串的每个字符
  for i in range(len(s)):
    # 如果两个字符串的某个字符不相等，则返回 false
    if s[i] != t[i]:
      return False

  # 如果两个字符串的每个字符都相等，则返回 true
  return True


def is_rotated_string(s, t):
  """
  判断两个字符串是否为旋转字符串。

  参数：
    s: 第一个字符串
    t: 第二个字符串

  返回：
    如果两个字符串是旋转字符串，则返回 true，否则返回 false
  """

  # 如果两个字符串的长度不相等，则返回 false
  if len(s) != len(t):
    return False

  # 循环比较两个字符串是否相等
  for i in range(len(s)):
    # 如果两个字符串相等，则返回 true
    if is_equal(s, t):
      return True

    # 将字符串 s 旋转一次
    s = rotate_string(s, 1)

  # 如果两个字符串都不相等，则返回 false
  return False


if __name__ == "__main__":
  # 测试用例
  test_cases = [
    ("abcde", "cdeab", True),
    ("abcde", "abdec", False),
    ("abc", "abc", True),
    ("abc", "cab", True),
    ("abc", "bca", True),
  ]

  # 运行测试用例
  for s, t, expected_result in test_cases:
    actual_result = is_rotated_string(s, t)
    print(f"s: {s}, t: {t}, expected_result: {expected_result}, actual_result: {actual_result}")