美文网首页
31. Next Permutation 下一个排列

31. Next Permutation 下一个排列

作者: sarto | 来源:发表于2022-03-22 17:42 被阅读0次

    题目

    给定一个整数数组 nums,数组中元素为 0 ~ 100 。找到下一个整数数组的排列,使得这个排列比当前排列大。
    [4,7,2,8,1,3] -> [4,7,2,8,3,1]
    [7,6,5,8,9,3,2] -> [7,6,5,9,2,3,8]
    [4,7,6,5,4,3] -> [5,3,4,4,6,7]
    [7,6,5,4,9,7,4,3,2] -> [7,6,5,7,9,2,3,4,4]

    解析

    找到重排序的规律:
    一个指针 flag 从 nums 尾部向头部移动,如果 nums[tail] < nums[tail-1],对掉两个数的位置,做一次从小到大的排序。
    特殊情况:
    如果遍历到末尾我们还没有找到这样一个解,直接将整个数组反序列

    1. 设置一个指针 指向倒数第二个数
    2. 比较 flag 和 flag - 1 如果大于等于,步进
    3. 如果小于,回溯回去,找到一个恰好比 flag 大的值,交换。记录 flag - 1
    4. 坐标内的数是递减的,将其整个反向。

    边界处理:
    [1,3,2]
    正向回溯时,如果最末尾的数仍然大于当前数,将最末尾的数和当前数做置换,例如 1 往后回溯,发现没有比 1 还小的数了,将末尾的数和 flag 对掉。

    伪代码

    for i = len-2;i >=0;i++
      if nums[i] < nums[i-1]
        for j:=i+2; i<len(num);j++
          if nums[j] <= nums[i]
              swap(nums[i], nums[j-1])
              break
      break
    reverse(nums, i+1,len)
    

    代码

    func nextPermutation(nums []int)  {
        var i int
        for i=len(nums)-2; i>=0; i-- {
            if nums[i] < nums[i+1] {
                j := i+2
                for j < len(nums) && nums[j] > nums[i]{
                    j++
                }
                swap(nums, i, j-1)
                break
            }
        }
        reverse(nums,i+1,len(nums)-1)
    }
    
    func reverse(nums []int, i int, j int) {
        for j>i {
            swap(nums, i ,j)
            i++
            j--
        }
    }
    
    func swap(nums []int, i int, j int) {
            nums[i] = nums[i] + nums[j]
            nums[j] = nums[i] - nums[j]
            nums[i] = nums[i] - nums[j]
    }
    
    image.png
    image.png

    同样的代码,两次效率不一样。

    相关文章

      网友评论

          本文标题:31. Next Permutation 下一个排列

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