leetcode31. 下一个排列

题目

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

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

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

以下是一些例子,输入位于左侧列,其相应输出位于右侧列。 1,2,3 → 1,3,2 3,2,1 → 1,2,3 1,1,5 → 1,5,1

解题思路

直接从数组的结尾遍历,如果nums[i] < nums[i+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
public void nextPermutation(int[] nums) {
if(nums.length == 1) return;
if(nums.length == 2) {swap(nums, 0, 1); return;}
int len = nums.length;
int index = len-2;
for(; index >= 0; index --) {
if(nums[index] < nums[index+1])
break;
}
if(index < 0) {
reserve(nums, 0, len-1);
return;
}
for(int i = len-1; i > index; i --) {
if(nums[i] > nums[index]) {
swap(nums, i, index);
reserve(nums, index+1, len-1);
return;
}
}
}
public void swap(int[] nums, int n, int m) {
int tmp = nums[n];
nums[n] = nums[m];
nums[m] = tmp;
}
public void reserve(int[] nums, int n, int m) {
while(n < m) swap(nums,n++,m--);
}

提交结果

成功
显示详情
执行用时 : 2 ms, 在Next Permutation的Java提交中击败了99.35% 的用户
内存消耗 : 38.2 MB, 在Next Permutation的Java提交中击败了80.97% 的用户