实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
思路:从后往前找到第一个nums[m]>nums[m-1]的位置m,分析此时m-1开始的序列保证都是降序排列的,那么继续从后往前找到第一个nums[j]>nums[m-1]的位置j,分析可知j>=m&&j<=n-1,找到j后调换索引m-1及j位置的值,并将m~n-1位置的值reverse即可,这样保证以最小的增量增大整个序列的值;
当然,如果整个序列为升序排列,那么直接reverse整个序列即可。
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int n = nums.size();
int m = n-1;
for(;m>=1;--m){
if(nums[m]>nums[m-1]){
break;
}
}
if(m>=1)
{
int j=n-1;
for(;j>=m;--j){
if(nums[j]>nums[m-1]){
swap(nums,j,m-1);
break;
}
}
reverse(nums,m,n-1);
}else{
reverse(nums,0,n-1);
}
return;
}
void swap(vector<int>& nums,int j,int m){
int temp = nums[j];
nums[j] = nums[m];
nums[m] = temp;
}
void reverse(vector<int>& nums,int i,int j){
while(i<j){
swap(nums,i,j);
i++;
j--;
}
return;
}
};