实现获取 下一个排列 的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列(即,组合出下一个更大的整数)。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须 原地 修改,只允许使用额外常数空间。
示例 1:
输入:nums = [1,2,3]
输出:[1,3,2]
示例 2:
输入:nums = [3,2,1]
输出:[1,2,3]
示例 3:
输入:nums = [1,1,5]
输出:[1,5,1]
示例 4:
输入:nums = [1]
输出:[1]
题解
- 思路:
- 字典排序下一个数,即找的最小的比原数大的数
- 562431 的字典排序下一个为 563124
- 从右往左查找,第一个 左数小于右数:2 和 4
- 此时 2 右边的数为降序,改变这些数的位置只会使原数变小
- 2 左边的数改变不如改变 2,否则就不是下一个数而是下下…个数
- 交换 2 和 3,交换后:5 6 3 4 2 1
- 2 右边与 2 最接近的大于 2 的数为 3,如果和 4 交换原数会更大
- 如果没有发现可以交换的数直接进行第四步
- 将 3 右边的数 升序排列,5 6 3 1 2 4
- 没有交换则全部进行升序排列
- 从右往左查找,第一个 左数小于右数:2 和 4
/**
* @param {number[]} nums
* @return {void} Do not return anything, modify nums in-place instead.
*/
var nextPermutation = function(nums) {
let index = nums.length - 2;
// 从右向左查找
while(index >= 0) {
// 第一个 左数 小于 右数(此时index右边的数一定是升序)
if(nums[index] < nums[index + 1]) break;
index--;
}
// 如果没有找到那么整个数组都是升序排列
// 在查找的 index 位置右边找到最接近且大于 index位上的数的数
if(index >= 0) { // 如果 index 不小于 0, 说明找到符合的数
// 查找 index 右边 最小但大于 index 位置上的数 的数
for(let i = nums.length - 1; i > index; i--) {
if(nums[i] > nums[index]) {
[nums[index], nums[i]] = [nums[i], nums[index]]; // 交换两个数
break;
}
}
}
let left = index + 1;
let right = nums.length - 1;
// 升序排列(原来一定是升序)
while(left < right) {
[nums[left], nums[right]] = [nums[right], nums[left]]; // 左右两个数对调
left++;
right--;
}
};