题目
实现获取下一个排列的函数,算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列,则将数字重新排列成最小的排列(即升序排列)。
必须原地修改,只允许使用额外常数空间。
以下是一些例子,输入位于左侧列,其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1
代码
/**
* 参考:
* https://leetcode-cn.com/problems/next-permutation/solution/xia-yi-ge-pai-lie-by-leetcode/
*/
public class Solution {
public static void main(String[] args) {
}
public static void nextPermutation(int[] nums) {
// 分成两步:
// 第一步:从后往前找到第一个下标为i的元素,a[i] < a[i + 1]
// 第二步:将a[i](除a[i]外)) 后面的所有的元素从小到大排序
int i = nums.length - 2;
// i >=0 确保不会出现数组越界异常
while (i >= 0 && nums[i] >= nums[i+1]) {
i--;
}
// nums[i] < nums[i+1]
if (i >= 0) {
int j = nums.length - 1;
while (j >= 0 && nums[j] <= nums[i]) {
j--;
}
// 此时 nums[j] > nums[i]
swap(nums, i, j);
}
// 将 i + 1 之后的元素排序
sort(nums, i + 1);
}
// 这里的排序算法和以往的都不一样,这是因为数组nums在下标start之后的元素是从大到小排序的
// 所以只需要反转即可将nums[start] 之后的元素从小到大排序
public static void sort(int[] nums, int start) {
int end = nums.length - 1;
while (start < end) {
swap(nums, start, end);
start++;
end--;
}
}
// 数值交换
private static void swap(int[] nums, int i, int j) {
nums[i] = nums[i] ^ nums[j];
nums[j] = nums[i] ^ nums[j];
nums[i] = nums[i] ^ nums[j];
}
}