美文网首页算法
算法:旋转数组

算法:旋转数组

作者: 某非著名程序员 | 来源:发表于2021-01-08 20:32 被阅读0次

    189. 旋转数组

    给定一个数组,将数组中的元素向右移动 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) 的 原地 算法。

    思路一

    1. 先处理k=1的时候,向右移动一次。
    2. 循环k次,及移动k次。空间复杂度O(1),时间复杂度O(k%n*n)
    func rotate(_ nums: inout [Int], _ k: Int) {
            var k1 = k
            if k == 0 {
                return
            }else if k>=nums.count{
                k1 = k%nums.count
            }
            
            for _ in 0..<k1 {
                var lastNum:Int = nums.last!
                var curNum:Int!
                
                for i in 0..<nums.count {
                    curNum = nums[i]
                    nums[i] = lastNum
                    lastNum = curNum
                }
            }
        }
    

    思路二

    这道题让我想起了《剑指offer》中的面试题42:翻转单词顺序 VS 左旋转字符串。
    基本思路就是先整体旋转,再局部旋转。先局部再整体也是可以的。

    //92.68% 34.43%
        func rotate(_ nums: inout [Int], _ k: Int) {
            var k1 = k
            if k == 0 {
                return
            }else if k>=nums.count{
                k1 = k%nums.count
            }
            let rotateIndex = nums.count-k1-1
            
            swapNums(&nums, 0, rotateIndex)
            swapNums(&nums, rotateIndex+1, nums.count-1)
            swapNums(&nums, 0, nums.count-1)
        }
        
        func swapNums(_ nums:inout [Int],_ i:Int,_ j:Int) {
            if i>=j {
                return
            }
            var i1 = i
            var j1 = j
            while i1<j1 {
                let temp = nums[i1]
                nums[i1] = nums[j1]
                nums[j1] = temp
                
                i1 += 1
                j1 -= 1
            }
        }
    

    总结

    1. 饮水思源,不求遇到新问题能做出来,只求遇到类似的问题能触类旁通。书不白看。

    相关文章

      网友评论

        本文标题:算法:旋转数组

        本文链接:https://www.haomeiwen.com/subject/hqyxaktx.html