Leetcode 31. 下一个排列【手动模拟求解】

问题描述

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

解题报告

手动模拟一下一个数组全排列的生成方式，做法是：
1.固定第一位，对剩余的数进行全排列；
2.当剩余的数全部全排列一遍后，下一个排列是：从剩余的数中选出第一个大于第一位的数，然后将该数换到第一位上，接下来对剩余的数进行从小到大排序。

所以我们求下一个排列时，只需要从当前排列的末尾开始遍历，找出第一个破坏单调递增顺序的那个数，然后将其与后面的数中第一个大于它的数进行交换，最后对它后面的数从小到大排序。

实现代码

• 参考官方

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int i = nums.size() - 2;
        while (i >= 0 && nums[i + 1] <= nums[i])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());
    }
};

• 使用库函数
  upper_bound(vec.begin(), vec.end(),target)找出数组 vec 中第一个大于 target 的数。
  lower_bound(vec.begin(), vec.end(),target)找出数组 vec 中第一个大于等于 target 的数。

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

• 更简单的代码
  next_permutation(vec.begin(), vec.end())可以直接生成当前排列的下一个排列，当前排列为最大排列时，生成最小的一个排列。

class Solution{
    public:
        void nextPermutation(vector<int>&nums){
            next_permutation(nums.begin(), nums.end());
        }
};

参考资料

[1] Leetcode 31. 下一个排列
[2] 题解区：官方答案