LeetCode31-下一个排列

发表于 | 分类于

题目链接

英文链接：https://leetcode.com/problems/next-permutation/

中文链接：https://leetcode-cn.com/problems/next-permutation/

题目详述

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

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

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

以下是一些例子，输入位于左侧列，其相应输出位于右侧列。
1,2,3 → 1,3,2
3,2,1 → 1,2,3
1,1,5 → 1,5,1

题目详解

全排列，找到下一个排列的四个步骤如下：

  1. 从后往前找到第一组位置，满足 nums[i] < nums[i + 1]（若找不到，说明已经是最后一个排列，即数组元素降序排列）。
  2. 从后往前找到第一个位置，满足 nums[i] < nums[j]。
  3. 交换 i、j 两处的值。
  4. 反转从位置 i + 1 开始到末尾的序列。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
public class LeetCode_00031 {

    public void nextPermutation(int[] nums) {
        int n = nums.length;
        int i = n - 2;
        // 1.从后往前找到第一组位置，满足 nums[i] < nums[i + 1]
        while (i >= 0 && nums[i] >= nums[i + 1]) {
            --i;
        }
        if (i >= 0) {
            int j = n - 1;
            // 2.从后往前找到第一个位置，满足 nums[i] < nums[j]
            while (nums[i] >= nums[j]) {
                --j;
            }
            // 3.交换 i、j 两处的值
            swap(nums, i, j);
        }
        // 4.反转从位置 i + 1 开始到末尾的序列
        reverse(nums, i + 1, n - 1);
    }

    private void swap(int[] a, int i, int j) {
        int t = a[i];
        a[i] = a[j];
        a[j] = t;
    }

    private void reverse(int[] a, int i, int j) {
        while (i < j) {
            swap(a, i++, j--);
        }
    }
}