LeetCode刷题|189旋转数组

这题是LeetCode编辑的初级算法中的一题,先来看一下题目描述吧。

题目描述

在这里插入图片描述
LeetCode地址:旋转数组

这个题目可以说是相当熟悉了,题目意思也很好理解,当时考研的时候教材里面就有这个题目,但是隔了这么久再次看到时,竟然还是想不起来咋做。于是我又开始了我最擅长的暴力解题法。首先介绍一下我自己的菜鸡解法。

我的解法一

我看到这个题第一个想法就是再创建一个数组,然后把每个元素的index往后挪三位放到新建的数组中。数组下标加k后如果超过数组长度就减一个数组长度。因为题目要求原地解决,不能有返回值,因此最后再把得到的数组复制到原数组。(可以说是非常蠢了)

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        if n<2:
            return
        new_nums = n*[0]
        i = 0
        for num in nums:
            if i+k<n:
                new_nums[i+k] = num
                i = i+1
            else:
                t = k
                while(i+t>=n):
                    t = t-n
                new_nums[i+t] = num
                i = i+1
        j = 0
        for new_num in new_nums:
            nums[j] = new_num
            j = j+1

但是上面说的思路还有一个问题,那就是k的值可能比数组长度还要大。因此就算减去一个数组长度n还是会越界,所以else部分加了一个循环,让下标和k的和循环减n,直到比数组长度要小就结束。这样就解决了所有问题。但是这个while循环做了很多无用功,白白耗费了很多时间,所以这个地方还是不太合理。

我的解法二

我所谓的解法二其实跟解法一差不多。解法一是将整个数组复制了一遍,这里我考虑只复制前面n-k个元素,将后面k个元素挪到前面k个位置,然后再将前面的n-k个元素往右挪k个单位。本质上跟上一种方法一样,直接看一下代码吧:

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        if n<2:
            return
        while(k>=n):
            k = k - n
        new_nums = (n-k)*[0]
        for i in range(0,n-k):
            new_nums[i] = nums[i]
        a = 0
        for j in range(n-k,n):
            nums[a] = nums[j]
            a = a+1
        for new_num in new_nums:
            nums[a] = new_num
            a = a+1

看了官方解法之后发现,我那个用while循环操作真的是太蠢了,直接一个取余操作秒杀我😭

class Solution {
    public void rotate(int[] nums, int k) {
        int n = nums.length;
        int[] newArr = new int[n];
        for (int i = 0; i < n; ++i) {
            newArr[(i + k) % n] = nums[i];
        }
        System.arraycopy(newArr, 0, nums, 0, n);
    }
}

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/rotate-array/solution/xuan-zhuan-shu-zu-by-leetcode-solution-nipk/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

看完官方这简洁的代码再看一下自己的,简直就是一坨屎。。。

我的解法三

这种解法就是教材上面的解法了,很多同学应该都比较熟悉。数组往右移动k位,先把数组整个做一个翻转,然后再把前k位翻转,最后再把后面n-k位翻转就得到最终的结果。

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        while(n<k):
            k = k-n
        self.reverse(nums,0,n)
        self.reverse(nums,0,k)
        self.reverse(nums,k,n)
        
    def reverse(self, num: List[int], m: int, n: int):
        for i in range(m,(n+m)//2):
            t = num[i]
            num[i] = num[m+n-i-1]
            num[m+n-i-1] = t

执行结果
可以看到,这个方法虽然实现了在原地操作数组,但是好像性能还是不太行。(可能是我代码写的不太行,思路应该没问题吧
最后还有一种官方比较复杂的解法,俺也没咋看懂,希望看懂的小伙伴可以给我解释一下😅

class Solution {
    public void rotate(int[] nums, int k) {
        int n = nums.length;
        k = k % n;
        int count = gcd(k, n);
        for (int start = 0; start < count; ++start) {
            int current = start;
            int prev = nums[start];
            do {
                int next = (current + k) % n;
                int temp = nums[next];
                nums[next] = prev;
                prev = temp;
                current = next;
            } while (start != current);
        }
    }

    public int gcd(int x, int y) {
        return y > 0 ? gcd(y, x % y) : x;
    }
}

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/rotate-array/solution/xuan-zhuan-shu-zu-by-leetcode-solution-nipk/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值