题目
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
思路
从后向前找到第一个a[i] < a[i + 1]
的位置,此时[i + 1, nums.length - 1]
为降序排列。再次从后向前查找第一个大于a[i]
的数,将这两个数交换,再将[i + 1, nums.length - 1]
反转即可。
代码
class Solution {
public void nextPermutation(int[] nums) {
if(nums.length == 1 || nums.length == 0)
return;
int pos = 0;
for (pos = nums.length - 2; pos >= 0; pos--) {
if(nums[pos] < nums[pos + 1])
break;
}
if(pos == -1) {
//原来是降序排列
Arrays.sort(nums);
return;
}
for (int i = nums.length - 1; i > pos; i--) {
if(nums[i] > nums[pos]) {
int tmp = nums[i];
nums[i] = nums[pos];
nums[pos] = tmp;
break;
}
}
int l = pos + 1, r = nums.length - 1;
while (l <= r) {
int tmp = nums[l];
nums[l] = nums[r];
nums[r] = tmp;
l ++;
r --;
}
}
}