LeetCode - 31. 下一个排列——数组

31. 下一个排列

题目描述

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

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

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

示例 1：

输入：nums = [1,2,3]
输出：[1,3,2]
示例 2：

输入：nums = [3,2,1]
输出：[1,2,3]
示例 3：

输入：nums = [1,1,5]
输出：[1,5,1]
示例 4：

输入：nums = [1]
输出：[1]

解题思路

1. 找到非降序的a[i - 1]
2. 寻找a[j]
3. 交换a[i - 1] 和 a[j]
4. 翻转a[i - 1]之后的元素
   

class Solution_31 {
    public void nextPermutation(int[] nums) {
        int i = nums.length - 2;
        //找到非降序的a[i - 1]
        while (i >= 0 && nums[i] >= nums[i + 1]){
            i--;
        }
        if (i >= 0){
            int j = nums.length - 1;
            while (j >= 0 && nums[i] >= nums[j])
                //寻找a[j]
                j--;
            //交换a[i - 1] 和 a[j]
            swap(nums,i,j);
        }
        //翻转a[i - 1]之后的元素
        reverse(nums,i + 1);
    }

    private void reverse(int[] nums, int start) {
        int left = start,right = nums.length - 1;
        while (left < right){
            swap(nums,left,right);
            left++;
            right--;
        }
    }

    private void swap(int[] nums, int i, int j) {
        int temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }

}