[LeetCode]31. 下一个排列

下一个排列

Posted by JinFei on September 27, 2021

题目描述

实现获取 下一个排列 的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列(即,组合出下一个更大的整数)。 如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。 必须 原地 修改,只允许使用额外常数空间。

Example 1:

输入:nums = [1,2,3] 输出:[1,3,2]

Example 2:

输入:nums = [3,2,1] 输出:[1,2,3]

Example 3:

输入:nums = [1,1,5] 输出:[1,5,1]

Example 4:

输入:nums = [1] 输出:[1]

Constraints:

  • 1 <= nums.length <= 100
  • 0 <= nums[i] <= 100

解题思路

  • 先找到不是递减的位置 i
  • 从后面开始,找到第一个不小于 nums[i] 的位置 j
  • 交换 nums[i], nums[j]
  • 交换 nums[i] 以及后面的序列

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[i] >= nums[j]){
                j--;
            }
            swap(nums[i], nums[j]);
        }
        reverse(nums.begin() + i + 1, nums.end());
    }
};