算法每日一刷

LeetCode算法题-31. 下一个排列(Swift)

2019-10-21  本文已影响0人  entre_los_dos

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/next-permutation
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目

实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。

如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。

必须原地修改,只允许使用额外常数空间。

以下是一些例子,输入位于左侧列,其相应输出位于右侧列。

1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

不明白题意,但是看解析是说。是这样
1、先找出最大的索引 k 满足 nums[k] < nums[k+1],如果不存在,就翻转整个数组;
2、再找出另一个最大索引 l 满足 nums[l] > nums[k];
3、交换 nums[l] 和 nums[k];
4、最后翻转 nums[k+1:]。

方法

 func nextPermutation(_ nums: inout [Int]) {
        
        if nums.count == 0 {
            return
        }
        var firstIndex = -1
        for i in stride(from: nums.count - 2, through: 0, by: -1) {
            if nums[i] < nums[i+1] {
                firstIndex = i
                break
            }
        }
        
        if firstIndex == -1 {
            reverseNums(&nums, 0, nums.count - 1)
            return
        }
        
        var secondIndex = -1
        for i in stride(from: nums.count - 1, through: 0, by: -1) {
            if nums[i] > nums[firstIndex] {
                secondIndex = i
                break
            }
        }

        swapNums(&nums, firstIndex, secondIndex)
        reverseNums(&nums, firstIndex + 1, nums.count - 1)
        return;

    }
    
    func reverseNums(_ nums: inout [Int], _ fromIndex: Int, _ toIndex: Int) {
        
        var fromIndex = fromIndex
        var toIndex = toIndex
        
        while fromIndex < toIndex {
            swapNums(&nums, fromIndex , toIndex)
            fromIndex += 1
            toIndex -= 1
        }
    }
    func swapNums(_ nums: inout [Int], _ fromIndex: Int, _ toIndex: Int) {
        let tmp = nums[fromIndex];
        nums[fromIndex] = nums[toIndex];
        nums[toIndex] = tmp;
    }
image.png
上一篇下一篇

猜你喜欢

热点阅读