算法

算法:旋转数组

2021-01-08  本文已影响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. 饮水思源,不求遇到新问题能做出来,只求遇到类似的问题能触类旁通。书不白看。
上一篇下一篇

猜你喜欢

热点阅读