题目大意
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
解题思路
下面的解题思路中统一以[1,3,2,4,3,2,1]举例。
首先,我们从后往前遍历数组,找到第一个降序的位置(因为该位置处是降序,因此可以从该位置的后续位置中找一个大于该位置的值,使得数组变大), 上述例子中,在index=2位置处于降序(4–>2为降序)。
已知index=2处是降序,那么index+1~length-1的元素是从后往前升序排列的,且至少有一个元素大于index位置的元素(因为index位置是降序位置,所以至少index+1位置的值大于index位置的值)。
我们从后往前遍历,找到第一个大于index位置的值,将其与index位置的值进行交换,上述例子中数组变为[1,3,3,4,2,2,1](加粗的数字表示交换过的)。
交换完毕后,在index+1~length-1位置同样不改变从后往前的升序规律(因为我们之前照的是第一个大于index位置的值进行的交换,所以交换过来后,左边都是大于该元素的,右边都是小于该元素的)。然后我们翻转index+1 ~length-1位置的元素,最终结果就是我们要找的下一个序列(翻转后,上述数组变为[1,3,3,1,2,2,4],加粗部分被翻转)。
class Solution {
public:
void nextPermutation(vector<int>& nums) {
if (nums.size() < 2)
return ;
int left = nums.size() - 2, right = nums.size() - 1;
//从后往前找到第一个降序的位置
while(left >= 0 && nums[left + 1] <= nums[left])
--left;
//如果一直都是升序的,表示数组是最大形式,直接反转为最小即可
if (left < 0){
reverse(nums, 0, nums.size() - 1);
return ;
}
//从后往前找到第一个大于nums[left]的位置
for (; right > left; --right){
if (nums[right] > nums[left])
break;
}
//交换两个位置的元素
int tmp = nums[left];
nums[left] = nums[right];
nums[right] = tmp;
//翻转left + 1 ~ length - 1位置的值
reverse(nums, left + 1, nums.size() - 1);
return ;
}
void reverse(vector<int> & nums, int left, int right){
int tmp = 0;
while (left < right){
tmp = nums[left];
nums[left] = nums[right];
nums[right] = tmp;
++left;
--right;
}
return ;
}
};