题目:
给定一个数组,将数组中的元素向右移动 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) 的原地算法。
解题思路(C):
以示例2为例:
第一个元素-1,移动后所在位置的下标是2=(0+2)%4
第一个元素-100,移动后所在位置的下标是3=(1+2)%4
第一个元素3,移动后所在位置的下标是0=(2+2)%4
第一个元素99,移动后所在位置的下标是1=(3+2)%4
复制一个临时数组,利用 (步长+k)%数组元素个数,取得移动后的元素位置,再对应复制回去
代码:
void rotate(int* nums, int numsSize, int k) {
int temp_array[numsSize];
int i = 0;
int j = 0;
for(i=0;i<numsSize;i++){
temp_array[i] = nums[i];
}
for(j=0;j<numsSize;j++){
nums[(j+k)%numsSize]=temp_array[j];
}
return *nums;
}
解题思路(python):
利用python的切片Slice。
什么是切片?截取一部分
用法格式(摘自参考文章3):
格式: [start:end:step]
• [:] 提取从开头(默认位置0)到结尾(默认位置-1)的整个字符串
• [start:] 从start 提取到结尾
• [:end] 从开头提取到end - 1
• [start:end] 从start 提取到end - 1
• [start:end:step] 从start 提取到end - 1,每step 个字符提取一个
• 左侧第一个字符的位置/偏移量为0,右侧最后一个字符的位置/偏移量为-1
以示例2为例,截取了元数组下标2到结尾,和第一个元素到1的数,拼接成一个新数组。
代码:
class Solution:
def rotate(self, nums: 'List[int]', k: 'int') -> 'None':
"""
Do not return anything, modify nums in-place instead.
"""
array_length = len(nums)
nums[:] = nums[array_length-k :] + nums[: array_length-k]
参考:
1、【Leetcode189-旋转数组Rotate Array】(C语言) https://blog.csdn.net/SoYangA/article/details/81542081
2、【leetcode】旋转数组【C、Python】http://www.mamicode.com/info-detail-2322977.html
3、Python 字符串切片(slice) https://www.cnblogs.com/cedrelaliu/p/6004352.html