题目:
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
示例 1:
输入:[1,2,3,4,5,6,7]
和 k = 3 输出:[5,6,7,1,2,3,4]
解释: 向右旋转 1 步:[7,1,2,3,4,5,6]
向右旋转 2 步:[6,7,1,2,3,4,5]
向右旋转 3 步:[5,6,7,1,2,3,4]
示例 2:
输入: [-1,-100,3,99]
和 k = 2
输出:
[3,99,-1,-100]
解释:
向右旋转 1 步:
[99,-1,-100,3]
向右旋转 2 步:
[3,99,-1,-100]
说明:
- 尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
- 要求使用空间复杂度为 O(1) 的原地算法。
思路一:
Python中有一种高级特性“切片”(slice)。这种切片操作经常用来取指定索引范围的操作,因为用循环很繁琐。在本题目中,
用切片操作再合适不过了。比如:[1,2,3,4,5,6,7],k = 3。新数组中需要将5,6,7放在最开始的位置,5位置的索引是数组长度减去k的值。将5后面的数放在新数组的开始位置,再将1到4放在后面。
代码一:
class Solution(object):
def rotate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: void Do not return anything, modify nums in-place instead.
"""
nums[:] = nums[len(nums)-k:]+nums[:len(nums)-k]
思路二:
最普遍的做法。就是按照上述示例解释的步骤依次进行。
进行K次循环,依次进行三个步骤:
①.读取最高位的数字。
②.将倒数第二位开始直到第一位的每个数字向后移动一位。
③.把第一步读取的最高位置的数字放在第一位。
代码二:
class Solution {
public void rotate(int[] nums, int k) {
int temp;
while(k > 0){
temp = nums[nums.length-1];
for(int i = nums.length-2;i>=0;i--){
nums[i+1] = nums[i];
}
nums[0] = temp;
k--;
}
}
}