leetcode31 下一个排列(Medium)

题目来源:leetcode31 下一个排列

题目描述:

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

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

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

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

解题思路:

找到a[i-1]<a[i]的最大下标,即最后一个上升的位置pos=i-1。然后从右到左边找到第一个比它大的位置,两者交换。再把pos后面的字母升序排序。

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int pos=-1;
        //找到最后一个上升的位置
        for(int i=nums.size()-1;i>=1;i--){
            if(nums[i-1]<nums[i]){
                pos=i-1;
                break;
            }
        }
        //没找到,说明现在已经是完全逆序了,改成升序。
        if(pos==-1) {
            sort(nums.begin(),nums.end());
            return;
        }
        //存在,则找到pos之后,最后一个比他大的位置,然后交换两者
        for(int i=nums.size()-1;i>pos;i--){
            if(nums[i]>nums[pos]) {
                swap(nums[i],nums[pos]);
                break;
            }
        }
        //然后把pos后的值升序排序
        sort(nums.begin()+pos+1,nums.end());
    }
};
posted @ 2020-07-16 21:19  拉里拉里啦啦  阅读(78)  评论(0编辑  收藏  举报