今天做这道题的时候,想了半天才想通,感觉自己太笨了,写个博客记录一下这道题。
一、题目
实现获取 下一个排列 的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列(即,组合出下一个更大的整数)。如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。必须 原地 修改,只允许使用额外常数空间。
例如:
输入:nums = [1,2,3]
输出:[1,3,2]
输入:nums = [3,2,1]
输出:[1,2,3]
二、解题思路
主要思路:从后往前遍历,依次判断当前元素是否小于其后相邻的元素,若找到符合这一条件的元素a,则记录该元素的位置,因为总是可以将其与后面较大元素交换来得到更大的整数。之后,从后往标记位置遍历,找到第一个大于标记元素的元素b,记录其位置。因为a后的元素一定是单调递减的,否则,元素a应该右移,因此若a已找到,则b一定存在并且是从元素末尾开始第一个大于a的数。找到a和b后,将二者交换,之后,再从a后第一个元素开始到末尾对该部分数组翻转,即可得到下一个更大的排列。
直观的理解:对xxxxxx2431,我们找到a=2,它是从后往前出现的第一个小于右边相邻数的数,即可以通过使用后面的数字将其替换,将它变为比原来更大的数。然后,从其后找到最小的比它大的数,即为b=3,交换a和b,此时已经得到比原来大的整数了,但是不是下一个比它大的数(下一个更大的排列)呢?未必!因为将高位数替换后,低位数的排序可能导致新得到的数不是下一个较大的数。拿这个例子来说,此时我们得到了xxxxxx3421,421比124大,所以需要将a后的数字翻转,才可得到最终的结果。那这个翻转操作又是为什么呢?因为原本a后的元素就是单调递减的,将a与其后最小的大于它的数交换后,a后必然也是单调递减的,因为最小的比它大的数总是从后往前依次找的,所以将a和b交换后,a总是小于b前的数,大于等于b后的数,不会破坏原本的单调性。所以当高位换成了较大的数后,下一个较大的数应该为高位以后数的最小的排列,所以才会有翻转操作,因为a后单调递减,翻转即可得到最小的排列,此时便得到了下一个更大的排列xxxxxx3124。
对于特殊情况,即原本输入就是最大的整数,此时a未找到,其下标为-1,所以同样地,还是翻转a后的数组,即翻转整个数组,得到最小的数。
三、代码
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int smaller=nums.size()-2,larger=0;
while(smaller>=0 && nums[smaller]>=nums[smaller+1]){
smaller--;
}
if(smaller>=0){ //只要smaller大于等于0,说明a存在,此时b必然存在!
while(larger>smaller && nums[larger]<=nums[smaller]){
larger--;
}
swap(nums[smaller],nums[larger]);
}
reverse(nums.begin()+smaller+1,nums.end());
}
};