给你一个数组,将数组中的元素向右轮转 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]
提示:
1 <= nums.length <= 105
-231 <= nums[i] <= 231 - 1
0 <= k <= 105
思路
当我们将数组的元素向右移动 k 次后,尾部k mod n个元素会移动至数组头部,其余元素向后移动 k%n个位置。
因此我们可以使用数组翻转的方法:
- 将所有元素翻转,这样尾部的 k % n 个元素就被移至数组头部。
- 然后我们再翻转 [0,(k%n)−1] 区间的元素和 [k%n,n−1] 区间的元素,即可得到最后的答案。
Java代码
暴力:
class Solution {
public void rotate(int[] nums, int k) {
int i,j;
int n=nums.length;
for( i=0,j=n-1;i<j;i++,j--){
int t=nums[j];
nums[j]=nums[i];
nums[i]=t;
}
k=k%n;
for(i=0,j=k-1;i<j;i++,j--){
int t=nums[j];
nums[j]=nums[i];
nums[i]=t;
}
for(i=k,j=n-1;i<j;i++,j--){
int t=nums[j];
nums[j]=nums[i];
nums[i]=t;
}
}
}
正解:
class Solution {
public void rotate(int[] nums, int k) {
int n=nums.length;
k=k%n;
fanzhuan(0,n-1,nums);
fanzhuan(0,k-1,nums);
fanzhuan(k,n-1,nums);
}
public void fanzhuan(int begin,int end,int[] nums){
for( int i=begin,j=end;i<j;i++,j--){
int t=nums[j];
nums[j]=nums[i];
nums[i]=t;
}
}
}