Python程序：检查右旋转后的数组是否为前n个自然数的递增或递减数组

假设我们有一个名为nums的数字列表，其中包含n个元素。我们必须检查是否可以通过任何次数的右移nums来创建一个包含前n个自然数的递增或递减列表，例如[1, 2, ..., n]或[n, n - 1, ..., 1]。

因此，如果输入类似于nums = [5,6,1,2,3,4]，则输出为True，因为我们可以将其右移四次以使数组变为[1,2,3,4,5,6]

为了解决这个问题，我们将遵循以下步骤：

• n := nums的长度
• 对于i从1到n-1循环：
  • 如果|nums[i - 1] - nums[i]|不等于1且|nums[i - 1] - nums[i]|不等于n-1，则
    • 返回False
• 返回True

示例

让我们看下面的实现来更好地理解：

def solve(nums):
   n = len(nums)
   for i in range(1, n):
      if abs(nums[i - 1] - nums[i]) != 1 and abs(nums[i - 1] - nums[i]) != n - 1:
         return False
   return True

nums = [5,6,1,2,3,4]
print(solve(nums))

输入

[5,6,1,2,3,4]

输出

True

Arnab Chakraborty

更新于：2021年10月14日

81 次浏览

开启你的职业生涯

完成课程获得认证

开始学习