31. 下一个排列 @Python @Go

原题

https://leetcode.cn/problems/next-permutation/description/

思路

先找出最大的索引 k 满足 nums[k] < nums[k+1]，如果不存在，就翻转整个数组；
再找出另一个最大索引 l 满足 nums[l] > nums[k]；
交换 nums[l] 和 nums[k]；
最后翻转 nums[k+1:]。

作者：powcai
链接：https://leetcode.cn/problems/next-permutation/solutions/3830/xia-yi-ge-pai-lie-by-powcai/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

复杂度

时间：O(n)
空间：O(1)

Python代码

class Solution:
    def nextPermutation(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """      
        def reverse(l, r):
            while l < r:
                nums[l], nums[r] = nums[r], nums[l]
                l += 1
                r -= 1 
            
        n = len(nums)
        for i in range(n-2, -1, -1):
            if nums[i] < nums[i+1]:
                # 已找到i, 再去找j
                for j in range(n-1, i, -1):
                    if nums[j] > nums[i]:
                        # 已找到j, 交换值
                        nums[i], nums[j] = nums[j], nums[i]
                        # 倒序排列
                        reverse(i+1, n-1)
                        return 
        # 没找到i, j, 对数组倒序排列
        reverse(0, n-1)
        

Go代码

func nextPermutation(nums []int) {
	// 匿名函数
	var reverse func(int, int)
	reverse = func(l int, r int) {
		for l < r {
			nums[l], nums[r] = nums[r], nums[l]
			l++
			r--
		}
	}
	n := len(nums)
	for i := n - 2; i > -1; i-- {
		if nums[i] < nums[i+1] {
			// 已找到i, 再去找j
			for j := n - 1; j > i; j-- {
				if nums[j] > nums[i] {
					// 已找到j, 交换值
					nums[i], nums[j] = nums[j], nums[i]
					// 倒序排列
					reverse(i+1, n-1)
					return
				}
			}
		}
	}
	// 没找到i, j, 对数组倒序排列
	reverse(0, n-1)
}