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
思路
- 感觉有点脑筋急转弯,看了题解明白操作,但是也还是不能为什么这样做一定是结果
- 从右往左扫描,找到升序的部分,记录
- 从右往左扫描,找到比记录大的第一个元素,记录
- 交换两个记录
- 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());
}
};