Leetcode-189-旋转数组

将包含* n* 个元素的数组向右旋转 *k *步。

例如,如果 n = 7 , k = 3,给定数组 [1,2,3,4,5,6,7] ,向右旋转后的结果为 [5,6,7,1,2,3,4]

注意:

尽可能找到更多的解决方案,这里最少有三种不同的方法解决这个问题。

提示:

要求空间复杂度为 O(1)


分析:

题意明显,就是数组内换位置。这里可能会有一个误区,空间复杂度要求为O(1),并不是说不能创建数组,可以创建,但不能让创建的数组超过常数级别的空间复杂度。另外K可以通过取模运算,避免重复的换位操作。各位小伙伴,八仙过海,各显神通,拿下这道题吧!


接下来给出 4 种Java的解法,推荐度由低到高。

解法一:

优点:不用创建新数组。
缺点:太耗时,平台可能因为耗时太长而不通过。

思路:向右旋转K步可以分解为做K次K=1的旋转,那么可以让每个元素都向右移1位,最后1位移到开头位置,总共做k次这种全局移位即可。

Java代码如下:

    public void rotate(int[] nums, int k) {
        if (nums.length == 0 || nums.length == 1) {
            return;
        }

        k = k % nums.length;

        int currentCache = 0;
        int targetCache = 0;
        for (int i = 0; i < k; i++) {
            for (int j = 0; j < nums.length; j++) {
                if (j == nums.length - 1) {
                    nums[0] = currentCache;
                    break;
                }

                if (j == 0) {
                    currentCache = nums[j];
                }
                targetCache = nums[j + 1];
                nums[j + 1] = currentCache;

                currentCache = targetCache;
            }
        }
    }

解法二:

优点:速度比解法一快很多。
缺点:需要新创建数组。

思路:这是最容易想到的方法,创建一个新数组,内容和原数组保持一致,然后将新数组的每一个元素按照脚标加k的新脚标放入原数组对应的位置。这样原数组就完成了新赋值。

Java解答如下:

    public void rotate(int[] nums, int k) {
        if (nums.length == 0 || nums.length == 1) {
            return;
        }


        int[] newNums = new int[nums.length];
        for (int i = 0; i < nums.length; i++) {
            newNums[i] = nums[i];
        }

        int index;
        for (int i = 0; i < nums.length; i++) {
            index = (i + k) % nums.length;
            nums[index] = newNums[i];
        }
    }

解法三:

优点:速度非常快。
缺点:需要创建一个比原数组长的新数组。

思路:由于是向右旋转k位,原数组的 [0, length - k) 个元素将向后移动k位,最终移到数组[k, length)的位置;而原数组的 [length - k, length) 个元素将会移动到数组的前面,占据了[0, k)的位置。
由此,创建一个新的长度是 length + k 的新数组,让新数组 [k, length + k) 的位置存放原数组的 [0, length);然后让新数组的 [0, k) 位存放原数组的 [length - k, length) 位。此时新数组的 [0, length) 就是原数组旋转之后的结果,再依次拷贝回原数组即可。

Java解答如下:

    public void rotate(int[] nums, int k) {
        int length = nums.length;
        if (length == 0 || length == 1) {
            return;
        }
        k %= length;
        int[] newnums = new int[length + k];
        System.arraycopy(nums, 0, newnums, k, length);
        System.arraycopy(nums, length - k, newnums, 0, k);
        System.arraycopy(newnums, 0, nums, 0, length);
    }

解法四:

优点:不用创建新数组,速度很快。
缺点:仅仅比解法三慢几毫秒。

思路:这里利用了一个小技巧。把原数组脚标为 0和length - 1的两个元素交换位置,再把 1和length - 2两个元素交换,依次类推直到整个数组顺序正好反转(原本{1,2,3,4,5,6}变成{6,5,4,3,2,1}),再把当前的数组 [0,k) 和 [k,length) 的位置分别进行反转,就这样,利用三次位置反转,就可以得到数组向右旋转k位的结果。
reverse()方法能把传入的数组arr的 [start, end] 的位置元素一一调换。

Java解答如下:

    public void rotate(int[] nums, int k) {
        int length = nums.length;
        if (length == 0 || length == 1) {
            return;
        }
        k %= length;
        reverse(nums, 0, length - 1);
        reverse(nums, 0, k - 1);
        reverse(nums, k, length - 1);
    }

    private void reverse(int[] arr, int start, int end) {
        while (start < end) {
            arr[start] = arr[start] + arr[end];
            arr[end] = arr[start] - arr[end];
            arr[start] = arr[start] - arr[end];
            start += 1;
            end -= 1;
        }
    }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容