美文网首页算法
31. 下一个排列

31. 下一个排列

作者: vbuer | 来源:发表于2018-08-20 15:51 被阅读156次

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

    示例

    输入位于左侧列,其相应输出位于右侧列。
    `1,2,3` → `1,3,2`
    `3,2,1` → `1,2,3`
    `1,1,5` → `1,5,1`
    

    代码

    class Solution {
    public:
    void nextPermutation(vector<int>& nums) {
            int k = -1;
            for (int i = nums.size() - 2; i >= 0; i--) {  //1、找到最大的下标k,使得nums[k] < nums[k + 1]
                if (nums[i] < nums[i + 1]) {
                    k = i;
                    break;
                }
            } 
            if (k == -1) {                               //如果没有知道这样一个下标,把整个序列反转即可
                reverse(nums.begin(), nums.end());
                return;
            }
            int l = -1;
            for (int i = nums.size() - 1; i > k; i--) {  //从后往前找,找到k之后满足nums[k] < nums[l]的最大的下标i。
                if (nums[i] > nums[k]) {
                    l = i;
                    break;
                } 
            } 
            swap(nums[k], nums[l]);   //交换nums[k] = nums[l]的值;
            reverse(nums.begin() + k + 1, nums.end());   //把数组下标从k+1到最后一个元素的序列反转。
        }
    };
    

    相关文章

      网友评论

        本文标题:31. 下一个排列

        本文链接:https://www.haomeiwen.com/subject/rgjliftx.html