旋转数组

旋转数组

题目描述

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

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

示例 1:

输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3,4]

示例 2:

输入：nums = [-1,-100,3,99], k = 2
输出：[3,99,-1,-100]

题目来源

力扣(LeetCode)

思路解析

方法一

右移k个位置 = > 将 i 移动到 （i + k）% nums.length ，但这里有个大坑，当k >= length 的时候，右移 k 其实是右移 k % length

空间复杂度 O(n)

package com.LeetCode;

/**
 * @author HelloWorld
 * @create 2021-04-22-22:09
 * @email 154803771@qq.com
 */
public class Test {
    public static void main(String[] args) {
        int[] nums = {1, 2, 3, 4, 5};
        rotate(nums, 3);
        for (int data : nums) {
            System.out.print(data + " ");
        }
    }

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

方法二

反转。先整体，再两次局部。但是尴尬的是Java 不可以直接对数组数组反转，需要手动实现

空间复杂度O(1)

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

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