美文网首页算法每日一刷
LeetCode算法题-31. 下一个排列(Swift)

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

作者: entre_los_dos | 来源:发表于2019-10-21 13:25 被阅读0次

    来源:力扣(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

    相关文章

      网友评论

        本文标题:LeetCode算法题-31. 下一个排列(Swift)

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