[Leetcode]下一个排列
题目描述
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
解题思路
- 首先我们从右边找到第一组两个连续的数字 a [ i ] a[i] a[i] 和 a [ i − 1 ] a[i-1] a[i−1],它们满足 a [ i ] > a [ i − 1 ] a[i]>a[i-1] a[i]>a[i−1]
- 想要创建比当前更大的排列,我们需要将数字 a [ i − 1 ] a[i-1] a[i−1]替换为其右侧区域数字中比它更大的数字 a [ j ] a[j] a[j],那么如果我们想找字典序的下一个排列, a [ j ] a[j] a[j]应该是 a [ i − 1 ] a[i-1] a[i−1]右侧区域中大于 a [ i − 1 ] a[i-1] a[i−1]中的所有数字中的最小值
- 交换 a [ i − 1 ] a[i-1] a[i−1]和 a [ j ] a[j] a[j],但此时的序列仍不是我们正在寻找的排列,我们还需要使 a [ i − 1 ] a[i-1] a[i−1]右边的数字按升序进行排列,以获得最小的排列
- 如果此时是降序排列,那么不存在下一个更大的序列,直接将数组反转reverse或者sort为升序序列
实现代码
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){//如果是0,说明是一个递减序列,不存在下一个更大的序列,那么此时进行升序排序
sort(nums.begin(),nums.end());
return;
}
int tmp=i-1;//记录此时的位置
while(i<len&&nums[i]>nums[tmp])//找到tmp右边大于nums[tmp]中的最小值
i++;
swap(nums[tmp],nums[i-1]);//交换
sort(nums.begin()+tmp+1,nums.end());//将tmp后的序列排序
}
};