给定一个数组,将数组中的元素向右移动 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]
解决方式一(反转数组):
Java版:
package com.lxf.demo02;
public class Rotate2 {
public static int[] rotate(int[] nums, int k) {
k %= nums.length;
//将整个数组反转
reverse(nums, 0, nums.length - 1);
//反转数组0到k-1的数
reverse(nums, 0, k - 1);
//反转数组k到最后一位的数
reverse(nums, k, nums.length - 1);
return nums;
}
/**
* 反转传入数组的指定区间顺序
* @param nums 传入的数组
* @param start 数组开始下标
* @param end 数组结束下标
*/
public static void reverse(int[] nums, int start, int end) {
while (start < end) {
int temp = nums[start];
nums[start] = nums[end];
nums[end] = temp;
start++;
end--;
}
}
public static void main(String[] args) {
int[] nums={1,2,3,4,5,6,7};
for (int i : rotate(nums, 3)) {
System.out.print(i);
}
}
}
C语言版:
#include<stdio.h>
//反转数组函数
void reverse_arrays(int start,int end,int* nums){
while(start<end){
int temp=nums[start];
nums[start]=nums[end];
nums[end]=temp;
start++;
end--;
}
}
//旋转数组k位函数
void rotate(int* nums, int numsSize, int k){
k=k%numsSize;
if(k==0){
return;
}
reverse_arrays(0,numsSize-1,nums);//将数组元素全部反转
reverse_arrays(0,k-1,nums);//将下标0至k-1的元素反转
reverse_arrays(k,numsSize-1,nums);//将下标k至numSize-1的元素反转
return;
}