LeetCode-031-下一个排列

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

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

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

示例 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]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/next-permutation

解题思路

从最后一个元素往前寻找第一个不满足降序排序的数, 记录位置
找到之后在该元素后面的元素中找到比该元素大但是尽可能小的数
两数交换后, 再将记录位置后面的序列翻转
如果找不到不满足降序排列的数, 则说明整个序列是降序的, 已经是最大排列
直接将整个序列翻转

代码

class Solution {
    public void nextPermutation(int[] nums) {
        // 从倒数第二个数开始
        int i = nums.length - 2;
        // 满足升序则继续往前
        while (i >= 0 && nums[i] >= nums[i + 1]) {
            i--;
        }
        // 如果i未越界则说明找到了不满足降序的数
        if (i >= 0) {
            // 从后往前找到第一个比nums[i]小的并交换位置
            int j = nums.length - 1;
            while (j > i && nums[j] <= nums[i]) {
                j--;
            }
            swap(nums, i, j);
        }
        // 翻转数组, 从 i + 1 开始
        reverse(nums, i + 1);
    }

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

    public void reverse(int[] nums, int start) {
        int i = start, j = nums.length - 1;
        while (i < j) {
            swap(nums, i++, j--);
        }
    }
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容