LeetCode-数组下一个排列

Posted 2020-12-11 Flix

题目描述

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

以下是一些例子，输入位于左侧列，其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

题目链接： https://leetcode-cn.com/problems/next-permutation/

思路

由于是要找下一个更大的排列，所以当前的排列变动之后，排列对应数值的变化越小越好，所以应该从后往前遍历。步骤如下：

• 从后往前遍历数组，找到第一个不满足从后往前升序的数字，假设数字的位置为 i-1，该数字为 nums[i-1];
• 将 [i, n) 范围内的数组升序排序，因为升序是最小的数字，n 为 nums 的长度；
• 如果 i>0 的话，从 i 开始遍历，找到第一个大于 nums[i-1] 的数字 nums[j]，并交换 nums[i-1] 与 nums[j]；

代码如下：

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        if(nums.empty()) return;

        int i = nums.size()-1;
        while(i>0 && nums[i-1]>=nums[i]){
            i--;
        }

        sort(nums.begin()+i, nums.end());
        for(int j=i; j<nums.size() && i>0; j++){
            if(nums[j]>nums[i-1]){
                swap(nums[j], nums[i-1]);
                break;
            }
        }
    }
};

参考

1、https://leetcode-cn.com/problems/next-permutation/solution/c-0msjian-dan-shi-xian-by-lcl-17/