leetcode 算法-旋转数组-189

leetcode 算法-旋转数组

leetcode 传送门

题目

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

示例 1:

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

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

解题思路

本提是根据题目构造旋转数组,非常简单的题目,但是却有很多巧妙的方法。

方法一:翻转法

思路非常简单,先将后k个元素和前len-k个元素进行一次翻转,然后再对整体数组进行一次翻转,就可以得到结果。

方法二:交换法
假设数组长度为n
由于旋转具有周期性,因此k>n时,和k%n一致,因此先执行取模操作,k=k%n,
观察这个旋转数组前后的差异,其实本质就做了一件事情,就是将数组后k个元素与前n-k个元素整体进行交换。

我们可以确定的是,后k个元素旋转完毕后一定在数组的前k个位置,因此我们可以先将后k个元素与到数组的前k个元素进行交换;
当第一次交换完毕后，前k个元素位置正确，后 n-k个元素中最后 k 个元素顺序错误，继续交换;
此时我们把原数组的后n-k个元素当做新数组重新进行上述步骤,之后新数组元素长度每次减k,直到新数组长度小于等于0结束。

翻转法代码

class Solution {
    public void rotate(int[] nums, int k) {
        int len =nums.length;
        k=k%len;
        reverse(nums,0 , len-1-k);
        reverse(nums, len-k, len-1);
        reverse(nums,0 , len-1);
    }
    public static void reverse(int[] nums ,int start,int end){
        int mid = (end-start)>>1;
        for (int i = 0; i <=mid; i++) {
            swap(nums, start+i,end-i);
        }
    }
    public static void swap(int[] nums ,int i,int j){
        int tmp = nums[i];
        nums[i]=nums[j];
        nums[j]=tmp;
    }
}

交换法代码

class Solution {
    public void rotate(int[] nums, int k) {
        int n = nums.length;
        k %= n;
        // 第一次交换完毕后，前 k 位数字位置正确，后 n-k 位数字中最后 k 位数字顺序错误，继续交换
        for (int start = 0; start < nums.length && k != 0; n -= k, start += k, k %= n) {
            for (int i = 0; i < k; i++) {
                swap(nums, start + i, nums.length - k + i);
            }
        }
    }
    public static void swap(int[] nums, int i, int j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}