【LeetCode题解】31. 下一个排列

题目

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

题解

(1)从后向前遍历,找到第一个不满足降序(不满足降序可能是相等,比如上面例子中的1,1,5)的元素;若初始序列全部是降序,则i为-1,直接跳转至(3);
(2)将该元素同它后面的元素中比它大的第一个元素交换;
(3)将该元素后的所有元素排列,使之成为最小的排列。

public void swap(int[] array, int i, int j) {
        int tmp = array[i];
        array[i] = array[j];
        array[j] = tmp;
    }
    public void nextPermutation(int[] nums) {
        if (nums == null || nums.length <= 1) {
            return;
        }
        //从后向前找到第一个不满足逆序的元素
        int i = nums.length - 2;
        while (i >= 0 && nums[i] >= nums[i + 1]) {
            //注意,这里有=,可以排除含有重复元素的情况
            i--;
        }
        //从i+1位置开始,向后查找比nums[i]大的最小元素
        if (i >= 0) {
            int j = i + 1;
            while (j  nums[i]) {
                j++;
            }
            swap(nums, i, j - 1); //交换,注意是同 j - 1交换
        }
        //将i之后的元素逆置(这里包含一种特殊情况,若该排列已经是字典序中的最大了,则下一个序列应该是最小字典序,因此,直接在这里逆置即可)
        int k = nums.length - 1;
        i++;
        while(i < k) {
            swap(nums, i, k);
            i++;
            k--;
        }
    }

《【LeetCode题解】31. 下一个排列》 image

找工作的同学扫码关注,每天一道面试算法题

    原文作者:码蹄疾
    原文地址: https://www.jianshu.com/p/66201f13a9a9
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞