轮转数组
需求
给你一个数组,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。
示例
输入: 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]
1
2
3
4
5
6
2
3
4
5
6
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]
1
2
3
4
5
2
3
4
5
思路
数组轮转有这么一个规律:当我们将数组的元素向右移动 k次后,尾部 k mod n个元素会移动至数组头部,其余元素向后移动 k mod n个位置。
简单说我们只需要:
- 首先对整个数组实行翻转,这样子原数组中需要翻转的子数组,就会跑到数组最前面。
- 这时候,从 k 处分隔数组,左右两数组,各自进行翻转即可。
数组长度=7,k=3的情况:操作 结果 原始数组 1 2 3 4 5 6 7 翻转所有元素 7 6 5 4 3 2 1 翻转[0 , k mod n - 1]区间的元素 5 6 7 4 3 2 1 翻转[k mod n , n - 1]区间的元素 5 6 7 1 2 3 4
代码
public void rotate(int[] nums, int k) {
int n=nums.length;
//先对k进行一次取模,防止k>n的情况
k%=n;
reverse(nums,0,n-1);//翻转全部元素
reverse(nums,0,k-1);//翻转k号元素左半部分
reverse(nums,k,n-1);//翻转k号元素右半部分
}
//数组没有翻转方法,只有StringBuilder才有,需要自己写
public void reverse(int[] nums,int start,int end){
while(start<end){
int temp=nums[start];
nums[start++]=nums[end];
nums[end--]=temp;
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
上次更新: 2023/12/29 11:32:56