LeetCode算法題-31. 下一個(gè)排列(Swift)

來(lái)源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/next-permutation
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán),非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。

題目

實(shí)現(xiàn)獲取下一個(gè)排列的函數(shù),算法需要將給定數(shù)字序列重新排列成字典序中下一個(gè)更大的排列。

如果不存在下一個(gè)更大的排列,則將數(shù)字重新排列成最小的排列(即升序排列)。

必須原地修改,只允許使用額外常數(shù)空間。

以下是一些例子,輸入位于左側(cè)列,其相應(yīng)輸出位于右側(cè)列。

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

不明白題意,但是看解析是說(shuō)。是這樣
1、先找出最大的索引 k 滿足 nums[k] < nums[k+1],如果不存在,就翻轉(zhuǎn)整個(gè)數(shù)組;
2、再找出另一個(gè)最大索引 l 滿足 nums[l] > nums[k];
3、交換 nums[l] 和 nums[k];
4、最后翻轉(zhuǎn) 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
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容