实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列(即组合出下一个更大的整数)。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
输入:nums = [1,1,5] 输出:[1,5,1]
一、两遍扫描
下一个排列意味着是字典序上比该排列还大的排列。“下一个”意味着用尽可能小的值和处在它右边的大值进行交换。这就分为如何找到靠右到小值,并找到相对的大值。
- 找右边尽可能小的值就是找右边第一个破坏降序规律的那个值,如果没有找到那么该排列其实就是最大的那个排列了。
- 较大值应该从右边开始找第一个大于“较小值”的。经过交换后,较大值右边就是降序,然后进行反转,降序变为升序之后即为下一个排列。
为什么交换后得到的依然是降序的?
- 本身找到就是破坏降序规律的较小值,那么交换后,对于原本比较大值大的值依然比较小值大,所以排在它前面的值依然是降序的。
- 对于较小值右边的值,因为条件是找到第一个大于较小值的较大值进行交换,也就是说原本较大值右边的值本来就比较小值小否则早就已经进行交换了。
代码:
void nextPermutation(vector<int>& nums) {
const auto n = nums.size();
int i = n - 2, j = n - 1;
while(i >= 0)
{
if(nums[i] < nums[i + 1])
{
break;
}
i--;
}
while(j >= 0 && i >= 0)
{
if(nums[j] > nums[i])
{
std::swap(nums[j], nums[i]);
break;
}
j--;
}
std::reverse(nums.begin() + i + 1, nums.end());
}
二、循环遍历
使用这个方法是因为我没有发现正确的规律,仅仅认为找到最靠右边的小数即可。因此使用了两层循环直接找最靠右的数。替换之后再进行排序得到最终的结果。这里我并没有看到其实应该还存在一个条件,即最靠右边的小数它的右边其实是降序排列的,而这个降序排列的条件将极大的简化算法的复杂度。
代码:
void nextPermutation(vector<int>& nums) {
const auto n = nums.size();
int pre = -1, cur = -1;
for(int i = n - 1; i >= 1; --i)
{
for(int j = i - 1; j >= 0; --j)
{
if(nums[i] > nums[j] && j > pre)
{
pre = j; cur = i;
}
}
if(i <= pre)
break;
}
if(pre != -1)
{
std::swap(nums[pre], nums[cur]);
std::sort(nums.begin() + pre + 1, nums.end());
return;
}
std::reverse(nums.begin(), nums.end());
}