LeetCode算法题-31. 下一个排列(Swift)
2019-10-21 本文已影响0人
entre_los_dos
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/next-permutation
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
不明白题意,但是看解析是说。是这样
1、先找出最大的索引 k 满足 nums[k] < nums[k+1],如果不存在,就翻转整个数组;
2、再找出另一个最大索引 l 满足 nums[l] > nums[k];
3、交换 nums[l] 和 nums[k];
4、最后翻转 nums[k+1:]。
方法
func nextPermutation(_ nums: inout [Int]) {
if nums.count == 0 {
return
}
var firstIndex = -1
for i in stride(from: nums.count - 2, through: 0, by: -1) {
if nums[i] < nums[i+1] {
firstIndex = i
break
}
}
if firstIndex == -1 {
reverseNums(&nums, 0, nums.count - 1)
return
}
var secondIndex = -1
for i in stride(from: nums.count - 1, through: 0, by: -1) {
if nums[i] > nums[firstIndex] {
secondIndex = i
break
}
}
swapNums(&nums, firstIndex, secondIndex)
reverseNums(&nums, firstIndex + 1, nums.count - 1)
return;
}
func reverseNums(_ nums: inout [Int], _ fromIndex: Int, _ toIndex: Int) {
var fromIndex = fromIndex
var toIndex = toIndex
while fromIndex < toIndex {
swapNums(&nums, fromIndex , toIndex)
fromIndex += 1
toIndex -= 1
}
}
func swapNums(_ nums: inout [Int], _ fromIndex: Int, _ toIndex: Int) {
let tmp = nums[fromIndex];
nums[fromIndex] = nums[toIndex];
nums[toIndex] = tmp;
}
image.png