题目描述
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
示例:
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
题目链接: https://leetcode-cn.com/problems/next-permutation/
思路
由于是要找下一个更大的排列,所以当前的排列变动之后,排列对应数值的变化越小越好,所以应该从后往前遍历。步骤如下:
- 从后往前遍历数组,找到第一个不满足从后往前升序的数字,假设数字的位置为 i-1,该数字为 nums[i-1](为什么要这么寻找呢?因为我们要从前面找到一个“小数”,从后面找到一个尽可能小的“大数”,然后将两者交换);
- 将 [i, n) 范围内的数组升序排序,因为升序是最小的数字,n 为 nums 的长度;
- 如果 i>0 的话,从 j 向后遍历,找到第一个大于 nums[i-1] 的数字 nums[j],并交换 nums[i-1] 与 nums[j];
代码如下:
class Solution {
public:
void nextPermutation(vector<int>& nums) {
if(nums.empty()) return;
int i = nums.size()-1;
while(i>0 && nums[i-1]<nums[i]){ // 找到第一个不满足从后往前升序的位置,该位置为 i-1
i--;
}
sort(nums.begin()+i, nums.end());
for(int j=i; j<nums.size() && i>0; j++){
if(nums[j]>nums[i-1]){
swap(nums[j], nums[i-1]);
break;
}
}
}
};
或者这样写:
class Solution {
public:
void nextPermutation(vector<int>& nums) {
if(nums.size()<2) return;
int i = nums.size()-1;
while(i>0 && nums[i-1]>=nums[i]) i--;
sort(nums.begin()+i, nums.end());
if(i<=0){ // 说明整个数组是降序排列,例如 [3,2,1]
sort(nums.begin(), nums.end());
return;
}
for(int j=i; j<nums.size(); j++){
if(nums[j]>nums[i-1]){
swap(nums[j], nums[i-1]);
break;
}
}
}
};
参考
1、https://leetcode-cn.com/problems/next-permutation/solution/c-0msjian-dan-shi-xian-by-lcl-17/