题目
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
我的解法(存在问题)
var nextPermutation = function(nums) {
const len = nums.length;
let i = len-2, t
while(i >= 0) {
if(nums[i] < nums[i+1]){
t = nums[i+1] - nums[i]
break
} else {
i -= 1
}
}
if(i < 0) {
// 已经是降序
return nums.sort((a, b) => a - b)
} else {
let k = i+1, temp
for(let j=i+2; j < len; j++) {
temp = nums[j] - nums[i]
if(temp > 0 && t > temp){
t = temp
k = j
}
}
t = [nums[i]].concat( nums.slice(i+1,k).sort((a, b) => a - b) )
temp = [nums[k]].concat( nums.slice(k+1, len).sort((a, b) => a - b) )
for(let j=0; j < temp.length; j++) {
nums[i+j] = temp[j]
}
temp = (i+len-k)
for(let j=0; j < t.length; j++) {
nums[temp+j] = t[j]
}
}
return nums
};
问题在于找到i,k后如何交换值的问题
看看别人的解法:
function nextPermutation(nums) {
let i = nums.length - 2; // 向左遍历,i从倒数第二开始是为了nums[i+1]要存在
while (i >= 0 && nums[i] >= nums[i + 1]) { // 寻找第一个小于右邻居的数
i--;
}
if (i >= 0) { // 这个数在数组中存在,从它身后挑一个数,和它换
let j = nums.length - 1; // 从最后一项,向左遍历
while (j >= 0 && nums[j] <= nums[i]) { // 寻找第一个小于 nums[i] 的数
j--;
}
[nums[i], nums[j]] = [nums[j], nums[i]]; // 两数交换,实现变大
}
// 如果 i = -1,说明是递减排列,如 3 2 1,没有下一排列,直接翻转为最小排列:1 2 3
let l = i + 1;
let r = nums.length - 1;
while (l < r) { // i 右边的数进行翻转,使得变大的幅度小一些
[nums[l], nums[r]] = [nums[r], nums[l]];
l++;
r--;
}
}
思路很清晰,[nums[i], nums[j]] = [nums[j], nums[i]]; // 两数交换,实现变大
这样交换两个值是厉害的,我没想过这样用的。