旋轉數組

旋轉數組

題目描述

給定一個數組，将數組中的元素向右移動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 [email protected]
 */
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];
            }
        }
    }
}
           

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;
    }
}