[Leetcode] Rotate Array 旋转数组

463 查看

Rotate Array

Rotate an array of n elements to the right by k steps.

For example, with n = 7 and k = 3, the array [1,2,3,4,5,6,7] is rotated to [5,6,7,1,2,3,4].

双数组

复杂度

时间 O(N) 空间 O(N)

思路

建立一个新数组,将位移过的数字放在新的数组中。

反转法

复杂度

时间 O(N) 空间 O(1)

思路

通过三次反转,我们可以很巧妙的实现旋转数组。首先我们将整个数组反转,然后将前k个数字反转,然后再将后面剩下的数字反转,就得到目标数组了。

注意

  • 反转数组最简单的方法是交换元素,而交换元素有至少三种方法(临时变量,相加相减,异或)

  • k可能大于数组长度,旋转不止一次,所以我们要先对k取余

代码

public class Solution {
    public void rotate(int[] nums, int k) {
        k = k % nums.length;
        reverse(nums, 0, nums.length-1);
        reverse(nums, 0, k-1);
        reverse(nums, k, nums.length-1);
    }
    
    private void reverse(int[] nums, int i, int j){
        while(i<j){
            swap(nums, i, j);
            i++;
            j--;
        }
    }
    
    private void swap(int[] nums, int i, int j){
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}

后续 Follow Up

Q:如果是向左旋转k位而不是向右呢?
A:还是同样的方法,只是之前在反转完整个数组后,将其前k个单独反转,后面的单独反转。而左旋时,我们是将其后k个单独反转,然后前面的单独反转。