实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
2,7,6,5,4,3,1→ 3,1,2,4,5,6,7
思路
由以上例子,我们不难看出,从后往前遍历数组,两两相互比较,找到第一个比后一个元素小的元素,比如最后一个例子中的 2 ,然后找到 2 之后的部分有序数组中比 2 的 数中的最小的一个(3)
交换它们的位置(2和3),最后将那部分有序数组(现在是3之后的)从小到大排序。
C++ 代码
class Solution {
public:
void nextPermutation(vector<int>& nums) {
if(nums.size() <= 1)
return ;
int n = nums.size() - 2;
while(n >= 0 && nums[n] >= nums[n + 1])
n--;
int m = nums.size() - 1;
if(n >= 0) {
while(m > n && nums[m] <= nums[n])
m--;
swap(nums[n], nums[m]);
}
sort(nums.begin() + n + 1, nums.end());
}
};