LeetCode 31. 下一个排列

题目描述

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/next-permutation

思路解析

该题的意思可能比较难理解。意思是找到一个排列，他组成的数字刚好大于给定的排列。

1. 从数组末尾开始遍历，如果数组从尾至头一直是递增的，则该排列为降序排列，直接输出该数组的升序排列即可（std::sort）
2. 若该排列从第\(i\)位开始至结尾是递减的，则我们需要将第\(i - 1\)位与其后所有数字中，刚好大于第\(i - 1\)位的数字进行交换。
3. 将第\(i\)位至末尾按升序排序。

代码实现

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int i;
        for(i = nums.size() - 1; i > 0; i--) {
            if(nums[i] > nums[i - 1])
                break;
        }
        if(i == 0) {
            sort(nums.begin(), nums.end());
            return;
        }
        int target = nums[i - 1];
        // curVal 记录 nums 第 i 位以后的刚好大于 nums[i - 1]的值
        // curInd 记录 curVal 对应的下标
        int curVal = nums[i];       // nums[i] > nums[i - 1]
        int curInd = i;
        for(int j = i; j < nums.size(); j++) {
            if(nums[j] > target && nums[j] < curVal) {
                curInd = j;
                curVal = nums[j];
            }
        }
        // 交换 nums[i - 1] 与 其后刚好大于 nums[i - 1]的数字
        nums[curInd] = nums[i - 1];
        nums[i - 1] = curVal;
        // 对 nums[i] 以后的数字排序
        sort(nums.begin() + i, nums.end());
        return;
    }
};