力扣 31. 下一个排列

题目说明

实现获取 下一个排列 的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列(即,组合出下一个更大的整数)。

如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。

必须 原地 修改,只允许使用额外常数空间。

  • 提示:
    • 1 <= nums.length <= 100
    • 0 <= nums[i] <= 100

示例

示例 1:

1
2
输入:nums = [1,2,3]
输出:[1,3,2]

示例 2:

1
2
输入:nums = [3,2,1]
输出:[1,2,3]

示例 3:

1
2
输入:nums = [1,1,5]
输出:[1,5,1]

示例 4:

1
2
输入:nums = [1]
输出:[1]

笔者理解

此题是一道数组算法问题,在力扣题库中被定义为中等题。

解法

当笔者阅读完此题后,采用两次扫描的解法,先找出第一个倒序不符合升序的元素,再寻找倒序第一个大于之前元素的元素,最后再翻转后面的元素,保证只比原数大一点,让我们来看看具体如何实现的吧。

实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class Solution {
/*
* 数组
* 两遍扫描
*/
public void nextPermutation(int[] nums) {
int i = nums.length - 2;
// 寻找第一个倒序不符合升序的元素
while (i >= 0 && nums[i] >= nums[i + 1]) {
i--;
}
if (i >= 0) {
int j = nums.length - 1;
// 寻找倒序第一个大于上面元素的元素
while (j >= 0 && nums[i] >= nums[j]) {
j--;
}
// 交换两元素
swap(nums, i, j);
}
// 翻转数组,保证只比原数组稍大
reverse(nums, i + 1);

}

// 交换两数
public void swap(int[] nums, int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}

// 从指定位置翻转数组
public void reverse(int[] nums, int start) {
int left = start, right = nums.length - 1;
while (left < right) {
swap(nums, left, right);
left++;
right--;
}
}
}

时间和空间效率还行,可见此解法还比较适合此题。

image-20210907134835157

总结

本题是今天的一题,难度是为中等,感兴趣的朋友都可以去尝试一下,此题还有其他更多的解法,朋友们可以自己逐一尝试。