31. 下一个排列

• LeetCode(力扣)

题目：

31. 下一个排列

难度：中等 🌟🌟🌟

---

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

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

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

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

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

---

题解：

首先，需要理解 字典序 。

举个例子，$[1,2,3]$ ，其字典序排列如下：

$[1,2,3][1,3,2][2,1,3][2,3,1][3,1,2][3,2,1]$

比如，$[2,3,1]$ 的下一个为 $[3,2,1]$ ，特别的对于 $[3,2,1]$ 的下一个为 $[1,2,3]$

算法流程，假设数组长度为 $n$ 的数据 $a$ ：

• 首先从最右边往左查找第一组满足 $a[i+1]>a[i]$ ，记录当前的位置 $Firstindex=i$，此时，可以知道，$[i+1,n-1]$ 必然为递减序列 （如果找不到满足条件的组，那么说明该序列已经是最大的序列，直接翻转整个序列即可）
• 然后在 $[i+1,n-1]$ 中从右往后找到第一个满足 $a[j]>a[Firstindex]$ 的元素，记录当前的位置 $Secondindex=j$
• 此时，交换 $a[Firstindex],a[Secondindex]$ ，可以证明 $[i+1,n-1]$ 必然为降序。最后，我们让区间 $[i+1,n-1]$ 变为升序即完成。

---

程序实现：

• Python

  class Solution:
      def nextPermutation(self, nums: List[int]) -> None:
          """
          Do not return anything, modify nums in-place instead.
          """
          def reverse(a,i,j):
              while i<j:
                  a[i],a[j] = a[j],a[i]
                  i += 1
                  j -= 1
  
          n = len(nums)
          first_idx,second_idx = -1,-1
  
          for i in range(n-1,0,-1):
              if(nums[i] > nums[i-1]):
                  first_idx = i-1
                  break
          if(first_idx == -1):
              reverse(nums,0,n-1)
              return 
  
          for i in range(n-1,first_idx,-1):
              if(nums[i] > nums[first_idx]):
                  second_idx = i 
                  break 
          nums[first_idx],nums[second_idx] = nums[second_idx],nums[first_idx]
  
          reverse(nums,first_idx+1,n-1)
          
          return 
  

• c++

  class Solution {
  public:
      void nextPermutation(vector<int>& nums) {
          int i=nums.size()-2;
          while(i>=0 && nums[i]>=nums[i+1])
          {
              i--;
          }
          if(i>=0)
          {
              int j=nums.size()-1;
              while(j>=0 && nums[j]<=nums[i]) j--;
              swap(nums[i], nums[j]);
          }
          reverse(nums.begin()+i+1,nums.end());
      }
  };
  

PS： 当然 c++ 也有强大的标准库函数 next_permutation

复杂度分析：

时间复杂度：$O(N)$ ，其中 $N$ 为给定序列的长度。我们至多只需要扫描两次序列，以及进行一次反转操作。

空间复杂度：$O(1)$ ，只需要常数的空间存放若干变量。

---

参考：

• LC-official
• LC-powcai