实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。1,2,3
→ 1,3,2
3,2,1
→ 1,2,3
1,1,5
→ 1,5,1
思路:
(1)从后向前遍历,找到第一个不满足降序(不满足降序可能是相等,比如上面例子中的1,1,5)的元素;若初始序列全部是降序,直接跳转至(3);
(2)将该元素同它后面的元素中比它大的第一个元素交换;
(3)将该元素后的所有元素排列,使之成为最小的排列。
public void nextPermutation(int[] nums) {
if (nums == null || nums.length < 2) {
return;
}
// 从后向前找到第一个不满足递减的元素
int i = nums.length - 2;
while (i >= 0 && nums[i] >= nums[i + 1]) {
// 注意,这里有=,可以排除含有重复元素的情况
i--;
}
// 从i+1位置开始,向后查找比nums[i]大的最小元素 由于之前的第一个条件可知 后面是递减的 所以最小元素离最远
if (i >= 0) {
int j = i + 1;
while (j < nums.length && nums[j] > nums[i]) {
j++;
} // 到达小于nums[i]的时候会退出
swap(nums, i, j - 1); // 交换 j要减1
}
// 将i之后的元素逆置(这里包含一种特殊情况,若该排列已经是字典序中的最大了,则下一个序列应该是最小字典序,因此,直接在这里逆置即可)
i++;
int end = nums.length - 1;
while (i < end) {
swap(nums, i, end);
i++;
end--;
}
}
public void swap(int[] array, int i, int j) {
int tmp = array[i];
array[i] = array[j];
array[j] = tmp;
}