31. 下一个排列
题目:
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
示例:
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
数学推导
字典序下一个排列其实就是让当前的排序变大一点点,幅度为最小的情况。想要变大,就一定是找到一个顺序排列的子排列,将大的移动到前面,这样就会变大。
那么为了变大一丢丢,就应该找到当前排列最后面的一个小顺序,然后将后面大一点点的数移到前面。
具体算法:
- 从后向前遍历数组,找到第一个 n u m s [ i ] > n u m s [ i − 1 ] nums[i] > nums[i - 1] nums[i]>nums[i−1]。这个操作保证了区间 [ i , e n d ) [i,end) [i,end)上的元素都是递减排列的。
- 再从后遍历区间 [ i , e n d ) [i, end) [i,end),找到第一个比 n u m s [ i − 1 ] nums[i - 1] nums[i−1]大的元素,交换这两个元素。
- 这时 [ i , e n d ) [i,end) [i,end)区间上的元素依然是递减排列的,因此可以直接翻转这个区间上的元素,这样就达到了只增大一点点的目的。
class Solution {
public void swap (int[] nums, int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
public void nextPermutation(int[] nums) {
int start = 0;
for (int i = nums.length - 1; i > 0; i--) {
if(nums[i] > nums[i - 1]) {
for (int j = nums.length - 1; j >= i; j--) {
if(nums[j] > nums[i - 1]) {
swap(nums, i - 1, j);
start = i;
break;
}
}
break;
}
}
for (int i = start; i < (start + nums.length) / 2; i++) {
swap(nums, i, nums.length - i + start - 1);
}
}
}
复杂度分析
-
时间复杂度: O ( n ) O(n) O(n)
最差情况需要遍历两遍数组 O ( 2 n ) = O ( n ) O(2n) = O(n) O(2n)=O(n)。
-
空间复杂度: O ( 1 ) O(1) O(1)
在原数组上操作,仅占用常数的额外空间。