LeetCode刷题笔记 31. 下一个排列

题目要求

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

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

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

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

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

算法背景

1. 从右至左，找到第一个小于右边的数X（6），其右边的序列是从右向左的增序；
2. 从右至左，找到第一个刚好大于X的数Y（7）；
3. 交换X，Y；
4. 将除第一个数外的序列倒序或从左向右排序。

C++解法

官方方法

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int id1=-1,id2;
        int submin;
        for(int i=nums.size()-1;i>0;i--){
            if(nums[i-1]<nums[i]){
                id1=i-1;
                id2=i;
                break;
            } 
        }
        if(id1==-1) sort(nums.begin(),nums.end());
        else{
            submin=nums[id1+1];
            for(int i=id1+1;i<nums.size();i++){
                if(nums[i]>nums[id1]&&nums[i]<submin) {
                    submin=nums[i];
                    id2=i;
                }
            }
            swap(nums[id1],nums[id2]);
            sort(nums.begin()+id1+1,nums.end());
        }
    }
};

作者：24shi-01fen-_00_01
链接：https://leetcode-cn.com/problems/next-permutation/solution/guan-fang-fang-fa-bu-tong-xie-fa-stl-next_permutat/

执行用时	内存消耗
12 ms	8.5 MB

解法二

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int id1=-1;
        for(int i=nums.size()-1;i>0;i--){
            if(nums[i-1]<nums[i]){
                id1=i-1;
                break;
            } 
        }
        if(id1==-1) sort(nums.begin(),nums.end());
        else{
            sort(nums.begin()+id1+1,nums.end());
            auto it = upper_bound(nums.begin() + id1, nums.end(), nums[id1]);
            swap(nums[id1],*it);
        }
    }
};

语法学习

C++ STL 中lower_bound() 和 upper_bound()讲解
彻底记住 lower_bound 和 upper_bound 功能和用法