题目介绍
- 实现获取 下一个排列 的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
- 如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
- 必须 原地 修改,只允许使用额外常数空间
力扣31题目链接:https://leetcode-cn.com/problems/next-permutation/
示例 1:
输入:nums = [1,2,3]
输出:[1,3,2]
示例2:
输入:nums = [3,2,1]
输出:[1,2,3]
示例3:
输入:nums = [1,1,5]
输出:[1,5,1]
方法一:暴力法
- 最简单的想法就是暴力枚举,我们找出由给定数组的元素形成的列表的每个可能的排列,并找出比给定的排列更大的排列。
- 但是这个方法要求我们找出所有可能的排列,这需要很长时间,实施起来也很复杂。因此,这种算法不能满足要求。
我们跳过它的实现,直接采用正确的方法。
复杂度分析
- 时间复杂度:O(n!),可能的排列总计有 n! 个。
- 空间复杂度:O(n),因为数组将用于存储排列。
方法二:一遍扫描
首先,我们观察到对于任何给定序列的降序排列,就不会有下一个更大的排列。例如,以下数组不可能有下一个排列:
[9, 5, 4, 3, 1]
这时应该直接返回升序排列。
所以对于一般的情况,如果有一个“升序子序列”,那么就一定可以找到它的下一个排列。具体来说,需要从右边找到第一对两个连续的数字 a[i] 和 a[i-1],它们满足 a[i]>a[i-1]。
所以一个思路是,找到最后一个的“正序”排列的子序列,把它改成下一个排列就行了。
不过具体操作会发现,如果正序子序列后没数了,那么子序列的“下一个”一定就是整个序列的“下一个”,这样做没问题;但如果后面还有逆序排列的数,这样就不对了。比如:
[1,3,8,7,6,2]
最后的正序子序列是[1,3,8],但显然不能直接换成[1,8,3]就完事了;而是应该考虑把3换成后面比3大、但比8小的数,而且要选最小的那个(6)。接下来,还要让6之后的所有数,做一个升序排列,得到结果:
[1,6,2,3,7,8]
代码演示如下:
// 思路:从后向前找到升序子序列,然后确定调整后子序列的最高位,剩余部分升序排列
public void nextPermutation1(int[] nums){
int n = nums.length;
// 1. 从后向前找到升序子序列,找到第一次下降的数,位置记为k
int k = n - 2;
while ( k >= 0 && nums[k] >= nums[k+1] )
k --;
// 找到k,就是需要调整部分的最高位
// 2. 如果k = -1,说明所有数降序排列,改成升序排列
if ( k == -1 ){
Arrays.sort(nums);
return;
}
// 3. 一般情况,k >= 0
// 3.1 依次遍历剩余降序排列的部分,找到要替换最高位的那个数
int i = k + 2;
while ( i < n && nums[i] > nums[k] )
i ++;
// 当前的i,就是后面部分第一个比nums[k]小的数,i-1就是要替换的那个数
// 3.2 交换i-1和k位置上的数
int temp = nums[k];
nums[k] = nums[i-1];
nums[i-1] = temp;
// 3.3 k之后的剩余部分变成升序排列,直接前后调换
int start = k + 1;
int end = n - 1;
while ( start < end ){
int tmp = nums[start];
nums[start] = nums[end];
nums[end] = tmp;
start ++;
end --;
}
}
复杂度分析
- 时间复杂度:O(N),其中 NN 为给定序列的长度。我们至多只需要扫描两次序列,以及进行一次反转操作。
- 空间复杂度:O(1),只需要常数的空间存放若干变量。