LeetCode 31 下一个排列

LeetCode 31 下一个排列

问题描述
实现获取下一个排列的函数，算法需要将给定数字序列重新排列成字典序中下一个更大的排列。
如果不存在下一个更大的排列，则将数字重新排列成最小的排列（即升序排列）。
必须原地修改，只允许使用额外常数空间。
以下是一些例子，输入位于左侧列，其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

数学推导

• 最小的字典序排列从左到右升序排列
• 非最小字典序排列从左到右先增后减
• 下一个排列需要将最大位置(前增后减)前一个元素(a)替换为其后大于(>a)的所有元素中最小的
• 最后将剩余元素重排序(升序)

执行用时：1 ms, 在所有 Java 提交中击败了98.72%的用户
内存消耗：38.3 MB, 在所有 Java 提交中击败了98.36%的用户

/**
 * @author CodeSPA
 * @date 2020/5/26 - 23:21
 */
/**思考:
 *1、要求使用常数的额外空间 O(NlogN)
 *  从右往左遍历，找到前一个小于后一个的情况
 * */
class Solution {
    public void nextPermutation(int[] nums) {
        if(nums.length<=1) return;

        int idx1, idx2;
        /*idx1为要替换的节点*/
        for(idx1=nums.length-2; idx1>=0; idx1--){
            if(nums[idx1]<nums[idx1+1])
                break;
        }
        if(idx1==-1){   //此时为最大序列(从左到右为降序)，应该返回最小序列
            Arrays.sort(nums);
            return;
        }

        /*要找到idx1右侧刚好大于idx1的元素*/
        for(idx2=nums.length-1; idx2>idx1; idx2--){
            if(nums[idx2]>nums[idx1])
                break;
        }

        /*将idx1处替换为idx2*/
        int tmp = nums[idx1];
        nums[idx1] = nums[idx2];
        nums[idx2] = tmp;
        /*将idx1右侧重新排序为升序*/
        Arrays.sort(nums, idx1+1, nums.length);
    }
}