【LeetCode刷题】旋转数组


题目链接:旋转数组

给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。

示例 1:
输入: 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:
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]

解题

题目叫“旋转数组”,题意却是平移数组,不过从反转数组的角度去考虑,可以发现一个规律。以$[1,2,3,4,5,6,7]$为例,可以看作是先反转,得到$[7,6,5,4,3,2,1]$,然后前3个再反转得到$[5,6,7,4,3,2,1]$,然后第4个到最后一个再反转,就得到了
$[5,6,7,1,2,3,4]$。

也就是对于一个数组来说,先整个反转,再反转第$0 - k$个,再反转第$k - length$个,就能得到目标结果。

class Solution {
    public void reserve(int[] nums, int start, int end) {
        for(int i=start, j=end-1; i<(end+start)/2; i++, j--) {
            // 借助位运算,直接交换变量值。
            nums[i] = nums[i]^nums[j];
            nums[j] = nums[j]^nums[i];
            nums[i] = nums[i]^nums[j];
        }
    }
    public void rotate(int[] nums, int k) {
        k = k % nums.length;
        if(k==0) return;
        
        reserve(nums, 0, nums.length);
        reserve(nums, 0, k);
        reserve(nums, k, nums.length);

    }
}

声明:楓の街|版权所有,违者必究|如未注明,均为原创|本网站采用BY-NC-SA协议进行授权

转载:转载请注明原文链接 - 【LeetCode刷题】旋转数组


Just For Fun...