31. 下一个排列
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/next-permutation
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题
将num变为下一个更大的排列;
解法1:
next_permutation(nums.begin(),nums.end());
解法2:
从右往左扫描nums,找到第一个前小后大的数,即nums[i]>nums[i-1],此时i需大于0;
接着从i开始往后扫描,找到比Nums[i-1]大的最小值nums[j],交换nums[i-1]与nums[j],最后将nums[i]即其后元素倒序排列即可;
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int n=nums.size();
int i=n-1;
while(i>0&&nums[i]<=nums[i-1]) i--; //i>0在前
//找到i,此时nums[i]>nums[i-1]
if(i>0){
int j=i;
while(nums[j]>nums[i-1]&&j<n-1) j++;
if(nums[j]<=nums[i-1]) //<=
swap(nums[j-1],nums[i-1]);
else swap(nums[j],nums[i-1]);
//把小的换到后面
}
Reverse(nums,i);
//把i和后面的数逆序
}
private:
void Reverse(vector<int>& nums,int i){
int j=nums.size()-1;
while(i<j)
swap(nums[i++],nums[j--]);
}
};
注意点
从后往前遍历寻找 i 时,需要将i>0条件放在前面;
找 j 分两种情况,找到比nums[i-1]小的nums[j],则交换j-1与i-1;
找到最后一个元素也比nums[i-1]大,交换最后一个元素;
总结
要换成下一个比自身大的数,需从右往左扫描;
若扫描过来为左边大于右边,则无论怎么换位都不会是更大的num;
所以当找到第一个i-1<i 时,这个i-1就是要被换到后面的值,因为要满足更大,所以要找右边比i-1大的最小值与i-1 交换;
交换后再将降序逆序为升序,则得到下一个比自身大的最小数,即next_permutation;