实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。1,2,3
→ 1,3,2
3,2,1
→ 1,2,3
1,1,5
→ 1,5,1
代码:
/*
从右往左扫描,找到第一个比后一个小的数。
把这个数和它之后第一个比它小的数交换
*/
class Solution {
public:
void swap(vector<int>& nums, int p, int q){
int t = nums[p];
nums[p] = nums[q];
nums[q] = t;
}
void reverse(vector<int>& nums, int p){
for(int i = p; i < (p+nums.size())/2; i ++)
swap(nums,i,nums.size()-i-1+p);
}
void nextPermutation(vector<int>& nums) {
bool flag = false;
int index;
for(int i = nums.size() - 1; i > 0; i --){
if(nums[i-1] < nums[i]){
flag = true;
index = i-1;
break;
}
}
if(flag){
for(int i = nums.size() - 1; i > index; i --){
if(nums[i] > nums[index]){
swap(nums,index,i);
break;
}
}
reverse(nums,index+1);
}
if(!flag)
reverse(nums,0);
}
};