leetcode菜狗入门 | 31. 下一个排列

下一个排列

题目描述

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

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

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

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

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

解题思路

• 从后往前遍历，如果 num[i] > num[i-1]，证明存在更大的。

• 从end 到 i 的部分是倒序排列的，所以我们依旧从后往前遍历，找到第一个比 num[i-1] 大的数的位置

• 将 i-1 与该位置的数交换，再把从 i 到 end 部分的数字顺序排列

• 如果整个数组已经是最大，就将其顺序排列即可

代码

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int flag = 0;
        // int cur = nums[nums.size()-1];
        int index = nums.size()-1;
        for(int i = nums.size()-1; i > 0; i--){
            if(nums[i] > nums[i-1]){
                while(index > i){
                    if(nums[index] > nums[i-1])
                        break;
                    index--;
                }
                int temp = nums[i-1];
                nums[i-1] = nums[index];
                nums[index] = temp;
                sort(nums.begin()+i, nums.end());
                flag = 1;
                break;
            }
        }
        if(flag == 0)
            sort(nums.begin(), nums.end());
    }
};