题目描述(简单难度)

转动数组,将数组的最后一个元素移动到开头,重复操作 k 次。

解法一

完全按照题目的意思,每次把末尾的元素移动到开头,当然移动前需要把所有元素后移一位,把第一个位置腾出来。

此外,如果 k 大于数组的长度,k 是等效于 k % n 的。举个例子,nums = [1 2 3]k = 4,操作 4 次和操作 4 % 3 = 1 次是一样的结果。

public void rotate(int[] nums, int k) {
    int n = nums.length;
    k = k % n;
    for (int i = 0; i < k; i++) {
        int temp = nums[n - 1];
        for (int j = n - 1; j > 0; j--) {
            nums[j] = nums[j - 1];
        }
        nums[0] = temp;
    }
}

时间复杂度:O(kn)

空间复杂度:O(1)

解法二

空间换时间,解法一中每个元素都需要移动 k 次,因为最后一个元素移到第一个位置的话,就进行了整体后移。不然的话,第一个位置原来的数就会被覆盖掉。

我们可以申请一个和原数组等大的数组,复制之前所有的值。这样的话,我们就可以随心所欲的在原数组上赋值了,不需要考虑值的丢失。

public void rotate(int[] nums, int k) {
    int n = nums.length;
    k = k % n;
    int[] numsCopy = new int[n];
    for (int i = 0; i < n; i++) {
        numsCopy[i] = nums[i];
    }

    //末尾的 k 个数复制过来
    for (int i = 0; i < k; i++) {
        nums[i] = numsCopy[n - k + i];
    }

    //剩下的数复制过来
    for (int i = k; i < n; i++) {
        nums[i] = numsCopy[i - k];
    }
}

时间复杂度:O(n)

空间复杂度:O(n)

解法三

上边的解法都是直接可以想到的,写完之后看了 官方 提供的解法,下边分享一下。

换一种题目的理解方式。

把数组看成一个圆环,而 k 的含义其实就是所有数字顺时针移动 k 个位置。

如果 k = 2,那么含义就是 1 放到 3 的位置,3 放到 5 的位置...

当然程序上,如果 1 放到 3 的位置,3 就会被覆盖了,我们应该用一个变量 pre 存储当前位置被占用的数字。

思想就是上边的了,代码的话可能会有不同的写法,下边的供参考。

public void rotate(int[] nums, int k) {
    int n = nums.length;
    k = k % n;
    if (k == 0) {
        return;
    }
    int count = 0; //记录搬移了多少个数字
    int start = 0;
    int current = start;
    int pre = nums[current];
    while (true) {
        do {
            //要移动过去的位置
            int next = (current + k) % n;
            //数字做缓存
            int temp = nums[next];
            //将数字搬过来
            nums[next] = pre;
            pre = temp;
            //考虑下一个位置
            current = next;
            count++;
            //全部数字搬移完就结束
            if (count == n) {
                return;
            }
        } while (start != current);
        //这里是防止死循环,因为搬移的位置可能会回到最开始的位置, 所以我们 start++, 继续搬移其他组
        start++;
        current = start;
        pre = nums[current];
    }
}

时间复杂度:O(n),每个数字仅搬移一次。

空间复杂度:O(1)

解法四

依旧是参考 官方 题解。

看具体的例子,1 2 3 4 5k = 2

转换后最终变成 4 5 1 2 3

其实可以分三步完成。

整体逆序 5 4 3 2 1

k 个再逆序 4 5 3 2 1

后边的再逆序 4 5 1 2 3

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

private void reverse(int[] nums, int start, int end) {
    while (start < end) {
        int temp = nums[start];
        nums[start] = nums[end];
        nums[end] = temp;
        start++;
        end--;
    }
}

时间复杂度:O(n)

空间复杂度:O(1)

解法一、解法二就是对题目最简单的理解,解法三和解法四是进一步对题目的剖析,很厉害。

results matching ""

    No results matching ""