【Leetcode】31.下一个排列 题解

题目

实现获取下一个排列的函数，算法需要将给定数字序列重新排列成字典序中下一个更大的排列。

如果不存在下一个更大的排列，则将数字重新排列成最小的排列（即升序排列）。

必须原地修改，只允许使用额外常数空间。

以下是一些例子，输入位于左侧列，其相应输出位于右侧列。
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 --;
        }
    }
}