随想
难顶哦,明明记得408备考时候做过类似的题,看来当时想的还是太简单了,真正调试起来很多地方都没有考虑到,比如k大于数组长度之类。还得真敲。
看评论还说是面试常见题。深码。
题目
给你一个数组,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。
提示:
1 <= nums.length <= 105
-231 <= nums[i] <= 231 - 1
0 <= k <= 105
进阶:
尽可能想出更多的解决方案,至少有 三种 不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗?
输入: nums = [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]
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]
最后一组测试没通过,那测试用例,可真离谱啊。最后一组测试用例
时隔月余,被我鼓捣出来了。并不太难。原本是在三个循环中修修补补——数组移位公式中加入取余运算,使其能够满足len<k的情况,可能把自己算混了。推到重来搞了一下,一遍过。
class Solution {
public void rotate(int[] nums, int k) {
int len = nums.length;
if(len <= k){
k = k%len;
}
int[] temp = new int[k];
for(int i=0; i<k; i++){
temp[i] = nums[len - k + i];
}
for(int i = len-k-1; i >= 0; i--){
nums[k+i] = nums[i];
}
for(int i=0; i<k; i++){
nums[i] = temp[i];
}
}
}