三段逆置 力扣189 轮转数组

力扣189 轮转数组

Posted by Hilda on March 13, 2025

189. 轮转数组

189. 轮转数组

给定一个整数数组 nums,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。

示例 1:

1
2
3
4
5
6
输入: 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]

示例 2:

1
2
3
4
5
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释: 
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]

提示:

  • 1 <= nums.length <= 105
  • -231 <= nums[i] <= 231 - 1
  • 0 <= k <= 105

进阶:

  • 尽可能想出更多的解决方案,至少有 三种 不同的方法可以解决这个问题。
  • 你可以使用空间复杂度为 O(1)原地 算法解决这个问题吗?

注:这道题有可能存在k > nums.length的情况。

例如:

输入

1
nums = [1,2], k = 3

预期结果

1
[2,1]

解法1:未通过原因(超出时间限制)

image-20250313144028171

1
2
3
4
5
6
7
8
9
10
11
12
13
class Solution {
    public void rotate(int[] nums, int k) {
        int len = nums.length;
        if (len == 1) return;
        for (int i = 0; i < k; i++) {
            int temp = nums[len - 1];
            for (int j = len - 2;j >= 0;j--){
                nums[j + 1] = nums[j];
            }
            nums[0] = temp;
        }      
    }
}

超出时间限制的原因是:

LeetCode 测试用例中,( n ) 和 ( k ) 可达 10510^510^5 级别。例如:

  • 输入:n = 10^5(数组长度),k = 10^5(旋转步数)。
  • 每次旋转移动 \(n-1 \approx 10^5\) 个元素,总操作次数为 \(k \cdot (n-1) \approx 10^5 \cdot 10^5 = 10^{10}\)次操作,远超 LeetCode 典型的 \(10^8\)操作时间限制

解法2:使用额外数组(非原地)

image-20250313150140209

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class Solution {
    public void rotate(int[] nums, int k) {
        //首先处理k 如果k > len 那么需要
        int len = nums.length;
        k = k % len;
        int[] res = new int[len];
        int rest = 0;
        for (int i = 0; i < len; i++) {
            if (i < k) {
                res[i] = nums[len - k + i];
            } else {
                res[i] = nums[rest++];
            }
        }
        // 赋值回去
        for (int j = 0; j < len; j++) {
            nums[j] = res[j];
        }    
    }
}

解法3:原地法(三段逆置)

首先优化k:k = k % len

然后做三件事:

  • 逆置整个数组
  • 逆置前k个数
  • 逆置剩下的len - k个数

比如1,2,3,4,5,6,7,如果k = 3

  • 逆置全部:7,6,5,4,3,2,1
  • 逆置前k个数:5,6,7,4,3,2,1
  • 逆置剩下7-3个数:5,6,7,1,2,3,4

代码:

image-20250313151204216

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
class Solution {
    public void rotate(int[] nums, int k) {
        //首先处理k 如果k > len 那么需要
        int len = nums.length;
        k = k % len;
        reverse(nums, 0, len - 1);
        reverse(nums, 0, k - 1);
        reverse(nums, k, len - 1); 
    }

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