31. 下一个排列

题目

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

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

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

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

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

思路

• 从末尾开始，找到第一个相邻的两个数a、b，a小于b。b
• 从末尾开始，找到第一个大于a的数c，a和c交换。c后面都是倒序的。
• 把c后面的数进行反转。
• 如果数组整个都是倒序的，整个数组反转。

代码

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

        int i = nums.size() - 2;

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

        if ( i >= 0 ) {
            int j = nums.size() - 1;

            while ( j > i && nums[j] <= nums[i] ) {
                --j;
            }

            swap( nums[i], nums[j] );
        }

        sort( nums.begin()+i+1, nums.end() );

        return;
    }
};