LeetCode:31 下一个排列

Posted 2021-06-16 成为大佬途中

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

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

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

样例
以下是一些例子，输入位于左侧列，其相应输出位于右侧列。

1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

 

举例说明：3 4 5 2 1 --> 3 5 1 2 4

1、从末端往前找，找到第一个下降的元素的位置k，即5 -> 4是下降的，因此找到4这个元素对应的位置1（从0开始）
2、从 [k + 1,n - 1] 区间中找到大于 nums[k] 的最小值，由于该区间具有单调性，可以用二分查找，找到了l == 5
3、交换k和l位置的元素，即交换 4 和 5，序列变成3 5 4 2 1
4、将[k + 1,n - 1]区间的元素逆序，变成3 5 1 2 4

 

代码：

class Solution {
public:
　　void nextPermutation(vector<int>& nums) {
　　　　int k = nums.size() - 1;
　　　　while (k > 0 && nums[k - 1] >= nums[k]) k --;    //从后往前找第一个非降序数字开头
　　　　if (k <= 0)　　　　　　　　//如果该序列为直接为降序，直接反转即可
　　　　{
　　　　　　reverse(nums.begin(), nums.end());
　　　　}else                   //否则，从非降序数字开头后找大于它的最后一个数，因为后面的数都为降序，最后一个即是大于它最接近的数
　　　　{
　　　　　　int t = k;  
　　　　　　while (t < nums.size() && nums[t] > nums[k - 1]) t ++;    //找到那个数
　　　　　　swap(nums[t - 1], nums[k - 1]);　　　　　　//交换
　　　　　　reverse(nums.begin() + k, nums.end());　　//将后面的数字反转
　　　　}
　　}
};