【技巧】Leetcode 31. 下一个排列【中等】

下一个排列

整数数组的一个 排列 就是将其所有成员以序列或线性顺序排列。

• 例如，arr = [1,2,3] ，以下这些都可以视作 arr 的排列：[1,2,3]、[1,3,2]、[3,1,2]、[2,3,1] 。

整数数组的 下一个排列 是指其整数的下一个字典序更大的排列。更正式地，如果数组的所有排列根据其字典顺序从小到大排列在一个容器中，那么数组的 下一个排列 就是在这个有序容器中排在它后面的那个排列。如果不存在下一个更大的排列，那么这个数组必须重排为字典序最小的排列（即，其元素按升序排列）。

• 例如，arr = [1,2,3] 的下一个排列是 [1,3,2] 。
• 类似地，arr = [2,3,1] 的下一个排列是 [3,1,2] 。
• 而 arr = [3,2,1] 的下一个排列是 [1,2,3] ，因为 [3,2,1] 不存在一个字典序更大的排列。

给你一个整数数组 nums ，找出 nums 的下一个排列。

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

示例 1：

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

示例 2：

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

示例 3：

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

解题思路

• 1、从右往左遍历数组，找到第一个不满足升序条件的元素，记为索引i。例：52341 num[i]=3 i=2
• 2、如果找不到这样的元素，说明数组是降序排列的，没有下一个更大的排列，将数组逆序即为字典序最小的排列。
• 3、否则，从右往左遍历数组，找到第一个比nums[i]大的元素，记为索引j。例：j=3,num[j]=4
• 4、交换nums[i]和nums[j]。例：52431
• 5、将索引i右边的元素逆序，以保证右侧是最小的排列。例：52431 i=2右边的逆序成13->52413

Java实现

public class NextPermutation {

    public void nextPermutation(int[] nums) {
        int n = nums.length;

        //Step 1:从右面找到第一个不满足升序条件的元素，即nums[i] < nums[i+1]
        int i = n - 2;
        while (i >= 0 && nums[i] >= nums[i + 1]) {
            i--;
        }
        
        // Step 2: 如果存在不满足升序条件的而元素，则i>=0
        // 从右往左遍历数组，找到第一个比nums[i]大的元素nums[j]
        if (i >= 0) {
            int j = n - 1;
            while (j >= 0 && nums[j] <= nums[i]) {
                j--;
            }
            
            // Step 3: 交换nums[i]和nums[j]
            swap(nums, i, j);
        }
        
        // Step 4: 将索引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;
    }

    /**
     * i+1一定是降序的
     * 翻转都会得到最小的序列
     * @param nums
     * @param start
     * @param end
     */
    private void reverse(int[] nums, int start, int end) {
        while (start < end) {
            swap(nums, start, end);
            start++;
            end--;
        }
    }

    public static void main(String[] args) {
        NextPermutation permutation = new NextPermutation();
        int[] nums1 = {1, 5, 1};
        permutation.nextPermutation(nums1);
        System.out.println(Arrays.toString(nums1));

        int[] nums2 = {5, 2, 3,4,1};
        permutation.nextPermutation(nums2);
        System.out.println(Arrays.toString(nums2));

        int[] nums3 = {5,6,4, 3, 2,1};//123465 next 123546
        permutation.nextPermutation(nums3);
        System.out.println(Arrays.toString(nums3));
    }
}

时间空间复杂度

• 时间复杂度：O(n)，其中n为数组nums的长度。因为需要遍历一次数组。

• 空间复杂度：O(1)。