题目
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
思路
1.该题需要复习
2.
【1 3 5 4 2】 —> 【1 4 5 3 2 】 —> 【1 4 2 3 5 】
综合以上,找某序列的下一个排列算法为:
1、找到右边“”最长“”的降序子序列,如【5,4,2】,此时pos定位到“3”的位置
2、从右到左遍历降序子序列【5,4,2】,找到第一个大于3的元素4,并交换
3、将第二步得到的子序列排序即可得到下一个排列;
实现
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int len = nums.size();
int i = len - 1;
while(i > 0 && nums[i] <= nums[i - 1]) {
i--;
}
if (i == 0) {
reverse(nums.begin(), nums.end());
return;
}
int temp = nums[i - 1];
for (int j = i; j < len; j++) {
if (j == len - 1 && nums[i - 1] < nums[j]) {
int temp = nums[j];
nums[j] = nums[i - 1];
nums[i - 1] = temp;
sort(nums.begin() + i, nums.end());
break;
}
if (nums[i - 1] < nums[j] && nums[i - 1] >= nums[j + 1]) {
int temp = nums[j];
nums[j] = nums[i - 1];
nums[i - 1] = temp;
sort(nums.begin() + i, nums.end());
break;
}
}
}
};