leetcode 31.下一个排列

leetcode 打码记录

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

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

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

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

解题思路
要求是给出下一个更大的排序,意思是一个比现在数字序列大但是是最小的序列。序列位数从左到右排列,因此可以从右到左搜索(从低位到高位搜索),如果当前位置 i 的右边存在更大的数,说明交换这两个数可以得到更大的序列,但此时应该保证与nums[i] 交换的数是 i 位置右边满足比nums[i] 大且是最小的数。最后交换完两个数之后对 i 位置后的数进行降序排列。
代码

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

        for(int i=nums.size()-2;i>=0;i--){
            int index = i;
            for(int j=i+1;jif(nums[j]>nums[i]){
                    if(nums[index]>nums[j] || index==i){  // 搜索找到满足大于nums[i]的最小的数的位置index
                        index = j;
                    }
                }
            }
            if(i!=index){
                int temp = nums[i];
                nums[i] = nums[index];
                nums[index] = temp;
                sort(nums.begin()+i+1, nums.end());     // 对i位置后序列进行降序
                return;
            }
        }

        sort(nums.begin(), nums.end());
        return;
    }


};

你可能感兴趣的:(数据结构与算法)