实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中,下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。必须原地修改,只允许使用额外常数空间。
输入:nums = [1,2,3]
输出:[1,3,2]
输入:nums = [3,2,1]
输出:[1,2,3]
思路:
我们可以将该问题形式化地描述为:给定若干个数字,将其组合为一个整数。如何将这些数字重新排列,以得到下一个更大的整数。如 123 下一个更大的数为 132。如果没有更大的整数,则输出最小的整数。
以 1,2,3,4,5,6 为例,其排列依次为:
123456
123465
123546
…
654321
可以看到有这样的关系:123456 < 123465 < 123546 < … < 654321。
class Solution {
public:
void nextPermutation(vector<int>& nums) {
int i=0;
for (i=nums.size()-2; i >= 0; -- i) { // 从后往前找到第一个相邻升序对
if (nums[i] < nums[i+1]) break;
}
if (i == -1) reverse(nums.begin(),nums.end()); // 无相邻升序对,必定为非递减序列
else {
for (int j=nums.size()-1; j >= i+1; -- j) { // 从后往前[i+1,end)找第一个大于a[i+1]的值
if (nums[i] < nums[j]) {
swap(nums[i],nums[j]); // 交换二者
reverse(nums.begin()+i+1,nums.end()); // 反转[i+1,end),变成升序
break;
}
}
}
}
};
- 这道题的巧妙之处就在于,for循环用来查找i的位置,int i = 0在外面定义好了,然后用for循环在恰当的时候break,此时就可以保留下i。
- 当交换完成后,「较大数」右边的数需要按照升序重新排列。这样可以在保证新排列大于原来排列的情况下,使变大的幅度尽可能小。也就是reverse的操作。
只需要STL一行代码:
class Solution {
public:
void nextPermutation(vector<int>& nums) {
next_permutation(nums.begin(),nums.end()); // STL功能和题目描述一致
}
};