国际空间研究组织 | ISRO CS 2008 |问题 66

ISRO CS 2008是一个印度国际空间研究组织(ISRO)进行的编程考试。问题66是其中的一道问题。

题目描述

给定一个整数数组，请编写一个程序，使它将该数组的元素向右移动k个位置，其中k是非负数。

示例

输入：[1, 2, 3, 4, 5], k = 2

输出：[4, 5, 1, 2, 3]

解释：

向右移动2个位置后，数组应该为[4, 5, 1, 2, 3]。

思路

一种常见的解法是，将整个数组旋转，再把前k个元素和后面的元素分别旋转一遍。具体步骤如下：

1. 将整个数组旋转
2. 将前k个元素旋转
3. 将后面的元素旋转

具体实现时，可以先实现一个旋转函数，再根据上述步骤实现。旋转函数可以直接使用Python中的切片操作，代码如下：

def rotate(nums: List[int], k: int) -> None:
    n = len(nums)
    k = k % n
    nums[:] = nums[n-k:] + nums[:n-k]

完整代码

from typing import List

def rotate(nums: List[int], k: int) -> None:
    n = len(nums)
    k = k % n
    nums[:] = nums[n-k:] + nums[:n-k]

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        k = k % n
        
        rotate(nums, n)
        rotate(nums, k)
        rotate(nums[k:], n-k)
        
        return nums

Markdown 代码片段

# 国际空间研究组织 | ISRO CS 2008 |问题 66

ISRO CS 2008是一个印度国际空间研究组织(ISRO)进行的编程考试。问题66是其中的一道问题。

## 题目描述

给定一个整数数组，请编写一个程序，使它将该数组的元素向右移动k个位置，其中k是非负数。

## 示例

输入：[1, 2, 3, 4, 5], k = 2

输出：[4, 5, 1, 2, 3]

解释：

向右移动2个位置后，数组应该为[4, 5, 1, 2, 3]。

## 思路

一种常见的解法是，将整个数组旋转，再把前k个元素和后面的元素分别旋转一遍。具体步骤如下：

1. 将整个数组旋转
2. 将前k个元素旋转
3. 将后面的元素旋转

具体实现时，可以先实现一个旋转函数，再根据上述步骤实现。旋转函数可以直接使用Python中的切片操作，代码如下：

```python
def rotate(nums: List[int], k: int) -> None:
    n = len(nums)
    k = k % n
    nums[:] = nums[n-k:] + nums[:n-k]

完整代码

from typing import List

def rotate(nums: List[int], k: int) -> None:
    n = len(nums)
    k = k % n
    nums[:] = nums[n-k:] + nums[:n-k]

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        k = k % n
        
        rotate(nums, n)
        rotate(nums, k)
        rotate(nums[k:], n-k)
        
        return nums