下一个排列

31. 下一个排列 - 力扣(LeetCode)

下一个排列_第1张图片

思路

1. 从右向左遍历数组,找到第一个满足 `nums[i] < nums[i + 1]` 的位置 `i`。
2. 如果找到了这样的位置 `i`,则再次从右向左遍历数组,找到第一个满足 `i < j` 且 `nums[i] < nums[j]` 的位置 `j`。
3. 将 `nums[i]` 和 `nums[j]` 交换位置。
4. 最后,反转从 `i + 1` 到数组末尾的部分,使得该部分变为升序排列。
5. 如果没有找到满足条件的 `i`,说明数组是降序排列的,此时直接反转整个数组,得到最小的排列。

代码

class Solution {
    public void nextPermutation(int[] nums) {
        int n = nums.length;
        int i = n - 2;
        //从后往前, 找到第一个满足 nums[i] < nums[i + 1] 的下标 i
        while (i >= 0 && nums[i] >= nums[i + 1]) {
            i--;
        }
        if (i >= 0) {
            int j = n - 1;
            //从后往前, 找到第一个满足 i < j 且 nums[i] < nums[j] 的下标 j
            while (i < j && nums[i] >= nums[j]) {
                j--;
            }
            //交换nums[i] nums[j]
            swap(nums, i, j);
        }
        //反转 i + 1 到 n - 1 这段数组 (如果没有符合条件的 i,说明数组是降序排列的,此时直接反转整个数组
        reverse(nums, i + 1, n - 1);
    }

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

    private void reverse(int[] nums, int start, int end) {
        while (start < end) {
            swap(nums, start, end);
            start++;
            end--;
        }
    }
}

你可能感兴趣的:(算法,算法,数据结构,java,力扣,排序算法)