31. 下一个排列
2018-08-20 本文已影响156人
vbuer
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
示例
输入位于左侧列,其相应输出位于右侧列。
`1,2,3` → `1,3,2`
`3,2,1` → `1,2,3`
`1,1,5` → `1,5,1`
代码
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int k = -1;
for (int i = nums.size() - 2; i >= 0; i--) { //1、找到最大的下标k,使得nums[k] < nums[k + 1]
if (nums[i] < nums[i + 1]) {
k = i;
break;
}
}
if (k == -1) { //如果没有知道这样一个下标,把整个序列反转即可
reverse(nums.begin(), nums.end());
return;
}
int l = -1;
for (int i = nums.size() - 1; i > k; i--) { //从后往前找,找到k之后满足nums[k] < nums[l]的最大的下标i。
if (nums[i] > nums[k]) {
l = i;
break;
}
}
swap(nums[k], nums[l]); //交换nums[k] = nums[l]的值;
reverse(nums.begin() + k + 1, nums.end()); //把数组下标从k+1到最后一个元素的序列反转。
}
};