21.1.4 31. 下一个排列

题目

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

示例 1:
输入:nums = [1,2,3]
输出:[1,3,2]

示例 2:
输入:nums = [3,2,1]
输出:[1,2,3]

示例 3:
输入:nums = [1,1,5]
输出:[1,5,1]

示例 4:
输入:nums = [1]
输出:[1]

提示:
1 <= nums.length <= 100
0 <= nums[i] <= 100

思路

  1. 感觉有点脑筋急转弯,看了题解明白操作,但是也还是不能为什么这样做一定是结果
  2. 从右往左扫描,找到升序的部分,记录
  3. 从右往左扫描,找到比记录大的第一个元素,记录
  4. 交换两个记录
  5. reverse反转降序的部分

代码

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int i = nums.size()-2;
        while(i >= 0 && nums[i] >= nums[i+1]) --i;
        if(i == -1) {
            reverse(nums.begin(), nums.end());
            return;
        }
        int rec = nums[i], recI = i;
        i = nums.size()-1;
        while(nums[i] <= rec) --i;
        swap(nums[recI], nums[i]);
        reverse(nums.begin() + recI + 1, nums.end());
    }
};
posted @ 2021-01-04 00:10  肥斯大只仔  阅读(45)  评论(0编辑  收藏  举报