旋转数组三种解法

问题

将一个有n个元素的数组向右旋转k步。例如，n= 7和k = 3，数组[1,2,3,4,5,6,7]被旋转到[5,6,7,1,2,3,4]。

你知道多少种不同的方法来解决这个问题吗?

解决方案

1.1 中间数组

我们可以直接创建一个新数组，然后将元素复制到新数组中。然后使用System.arraycopy()更改原来的数组。

public static void rotate1(int[] nums, int k) {
        // 取余
        if (k > nums.length) {
            k = k % nums.length;
        }
        // 中间数组result
        int[] result = new int[nums.length];

        for (int i = 0; i < k; i++) {
            result[i] = nums[nums.length - k + i];
        }
        int j = 0;
        for (int i = k; i < nums.length; i++) {
            result[i] = nums[j];
            j++;
        }
        System.arraycopy(result, 0, nums, 0, nums.length);
    }

空间和时间复杂度都是O(n)。

1.2 冒泡旋转

我们能在O(1)空间中做这个吗?下面的解决方案类似于冒泡排序。

public static void rotate2(int[] arr, int order) {
        if (arr == null || order < 0) {
            throw new IllegalArgumentException("Illegal argument!");
        }

        for (int i = 0; i < order; i++) {
            for (int j = arr.length - 1; j > 0; j--) {
                int temp = arr[j];
                arr[j] = arr[j - 1];
                arr[j - 1] = temp;
            }
        }
    }

但是，时间复杂度是O(n*k)。

1.3 反转法

我们能在O(1)空间和O(n)时间内做这个吗?下面的解决方案是这样的。

假设已知1 2 3 4 5 6和旋转次数是2。其基本思想是:

1. 将数组分成两部分:1,2,3,4和5,6

2. 旋转第一部分:4,3,2,1,5,6

3. 旋转第二部分:4,3,2,1,6,5

4. 旋转整个数组:5,6,1,2,3,4

public static void rotate3(int[] arr, int order) {
        order = order % arr.length;

        if (arr == null || order < 0) {
            throw new IllegalArgumentException("Illegal argument!");
        }

        int a = arr.length - order;
        reverse(arr, 0, a - 1);
        reverse(arr, a, arr.length - 1);
        reverse(arr, 0, arr.length - 1);
    }

    public static void reverse(int[] arr, int left, int right) {
        if (arr == null || arr.length == 1) return;

        while (left < right) {
            int temp = arr[left];
            arr[left] = arr[right];
            arr[right] = temp;
            left++;
            right--;
        }
    }

测试代码

public static void main(String[] args) {
        int[] arr1 = {1, 2, 3, 4, 5, 6, 7};
        rotate1(arr1, 3);
        System.out.println(Arrays.toString(arr1));

        int[] arr2 = {1, 2, 3, 4, 5, 6, 7};
        rotate2(arr2, 3);
        System.out.println(Arrays.toString(arr2));

        int[] arr3 = {1, 2, 3, 4, 5, 6, 7};
        rotate3(arr3, 3);
        System.out.println(Arrays.toString(arr3));
    }

结果：

[5, 6, 7, 1, 2, 3, 4]
[5, 6, 7, 1, 2, 3, 4]
[5, 6, 7, 1, 2, 3, 4]

进程完成，退出码 0