题目:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。 1,2,3 → 1,3,2 3,2,1 → 1,2,3 1,1,5 → 1,5,1
抛砖引玉
思路:
- 1,2,3 → 1,3,2 数组从后到前递减组合起来是最小的排列
- 3,2,1 → 1,2,3 数组从后到前递增组合起来是最大的排列
题目要求如果存在下一个更大排列则输出、否则输出最小组合:
- 数组从后到前遍历存在递减片段则说明存在更大排列
- 例如:1,3,2 → 2,1,3,其中3到1递减,则将1与2替换然后翻转3,1 -> 2,1,3
- 例如:4,5,2,6,3,1 → 4,5,3,1,2,6,其中6到2递减,则将2与3替换然后翻转6,2,1 -> 4,5,3,1,2,6
- 数组从后到前递增都为递增则整体翻转数组则得到最小组合
/**
* @param {number[]} nums
* @return {void} Do not return anything, modify nums in-place instead.
*/
var nextPermutation = function (nums) {
let len = nums.length,
index = len - 2
// 特殊情况如果nums小于两个元素直接返回
if (len < 2) return nums
while (index >= 0 && nums[index] >= nums[index + 1]) {
index--
}
// 倒序遍历中存在递减片段
if (index >= 0) {
let i = len - 1;
// 将递减片段的第一个元素替换到倒序遍历开始的递增片段中
while (i >= 0 && nums[index] >= nums[i]) {
i--
}
[nums[i], nums[index]] = [nums[index], nums[i]]
}
// 翻转已经遍历的递增片段,如果nums整体倒序递增则翻转得到最小组合
reverse(index + 1)
// 翻转指定位置之后的元素
function reverse (index) {
let start = index,
end = len - 1;
while (start < end) {
[nums[start], nums[end]] = [nums[end], nums[start]]
start++
end--
}
}
return nums
};
博客: 前端小书童
每天的每日一题,写的题解会同步更新到公众号一天一大 lee 栏目 欢迎关注留言