【LeetCode面试经典150题】189. 轮转数组

奕_Raku

已于 2024-05-16 14:33:29 修改

阅读量186

点赞数 1

分类专栏： LeetCode 文章标签： leetcode 面试算法

于 2024-05-15 21:26:35 首次发布

本文链接：https://blog.csdn.net/qq_29536773/article/details/138904103

版权

LeetCode 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

一、题目

189. 轮转数组 - 力扣（LeetCode）
给定一个整数数组 nums，将数组中的元素向右轮转 k 个位置，其中 k 是非负数。

二、思路

读完题觉得简单，不就是轮转嘛，我只要每次把最后一个元素转到开头，转 k 次不就结了？于是写了解法一，直接超时光荣牺牲；
然后懵了，重新开始想，我可以先新建一个数组，把 nums[k] 往后的数组单独提出来装进新数组中，装完后从 nums[0] 开始把原数组装进新数组中，于是有了方法二；
这里要注意，为什么 k 要取与 len 的余：想象一个数组长度为三，k=5 其实等价于 k=2；
看了看题解，理解之后写了方法三，确实很巧妙；

// 原地址：https://leetcode.com/problems/rotate-array/discuss/54250/Easy-to-read-Java-solution

数组 = "----->-->"; k = 3
要求 = "-->----->";

反转 "----->-->" ，可以得到 "<--<-----"
反转 "<--" ，可以得到 "--><-----"
反转 "<-----" ，可以得到 "-->----->"

三、解法

[超时]解法一

class Solution {
    public void rotate(int[] nums, int k) {
        int len = nums.length;
        while (k != 0) {
            for (int tail = len - 1; tail > 0; tail--) {
                int tmp = nums[tail - 1];
                nums[tail - 1] = nums[tail];
                nums[tail] = tmp;
            }
            k--;
        }
    }
}

解法二

class Solution {
    public void rotate(int[] nums, int k) {
        int len = nums.length;
        k %= len; // <- 若 k>len 时，如 k=5 len=3 实际只需要转2位
        int[] ret = new int[len];
        int j = 0;
        for (int i = len - k; i < len; i++) {
            ret[j++] = nums[i];
        }
        int m = 0;
        while (j < len) {
            ret[j++] = nums[m++];
        }
        for (int n = 0; n < len; n++) {
            nums[n] = ret[n];
        }
    }
}

解法三

class Solution {
    // 轮转操作
    public void rotate(int[] nums, int k) {
        int len = nums.length;
        k %= len; // <- 若 k>len 时，如 k=5 len=3 实际只需要转2位
        // 先整体反转
        reverse(nums, 0, len - 1);
        // 反转左半部分
        reverse(nums, 0, k - 1);
        // 反转右半部分
        reverse(nums, k, len - 1);
    }
    // 反转操作
    public void reverse(int[] nums, int start, int end) {
        while (start < end) {
            int tmp = nums[start];
            nums[start++] = nums[end];
            nums[end--] = tmp;
        }
    }
}