【LeetCode-数组】下一个排列

2021-05-11 20:29

阅读:583

标签:--   c++   for   假设   http   while   算法   描述   return   

题目描述

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

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

题目链接: https://leetcode-cn.com/problems/next-permutation/

思路

由于是要找下一个更大的排列,所以当前的排列变动之后,排列对应数值的变化越小越好,所以应该从后往前遍历。步骤如下:

  • 从后往前遍历数组,找到第一个不满足从后往前升序的数字,假设数字的位置为 i-1,该数字为 nums[i-1];
  • 将 [i, n) 范围内的数组升序排序,因为升序是最小的数字,n 为 nums 的长度;
  • 如果 i>0 的话,从 i 开始遍历,找到第一个大于 nums[i-1] 的数字 nums[j],并交换 nums[i-1] 与 nums[j];

代码如下:

class Solution {
public:
    void nextPermutation(vector& nums) {
        if(nums.empty()) return;

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

        sort(nums.begin()+i, nums.end());
        for(int j=i; j0; j++){
            if(nums[j]>nums[i-1]){
                swap(nums[j], nums[i-1]);
                break;
            }
        }
    }
};

参考

1、https://leetcode-cn.com/problems/next-permutation/solution/c-0msjian-dan-shi-xian-by-lcl-17/

【LeetCode-数组】下一个排列

标签:--   c++   for   假设   http   while   算法   描述   return   

原文地址:https://www.cnblogs.com/flix/p/13149844.html


评论


亲,登录后才可以留言!