给你一个数组,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。
示例 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]
示例 2:
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]
https://leetcode.cn/problems/rotate-array/
( 1 ) (1) (1)创建一个和原数组一样大的数组,将原数组中的元素从倒数第k个位置开始依次放入新数组。
( 2 ) (2) (2)超出边界从0开始,直到新数组填满。
( 1 ) (1) (1)先将整个数组反转
( 2 ) (2) (2)以第k个元素位置为分界线,分别反转两个子数组。
( 3 ) (3) (3)比如[1,2,3,4,5,6],k=4。第一次反转结果为[6,5,4,3,2,1],然后以下标3为分界线分成两个子数组[6,5,4,3],[2,1],分别反转之后[3,4,5,6,1,2]
时间复杂度O(n),额外数组空间复杂度O(n),不使用O(1)
Java
class Solution {
public void rotate(int[] nums, int k) {
if (nums.length == 1 || k%nums.length==0)return;
int[] res = new int[nums.length];
int l = nums.length - k%nums.length;
for(int i=0;i<nums.length;i++) {
res[i] = nums[i];
}
for(int i=0;i<nums.length;i++) {
nums[i] = res[l];
l = (l+1)%nums.length;
}
}
}
C++
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int n = nums.size();
k = k%n;
if (k==0) k=n;
reverse(nums.begin(), nums.end());
reverse(nums.begin(), nums.begin()+k);
reverse(nums.begin()+k, nums.end());
}
};
1.记住这个技巧