LeetCode31 下一个排列

题目

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

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

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

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

 提示: 
 1 <= nums.length <= 100 
 0 <= nums[i] <= 100 

方法

双指针法

找到下一个排列需要三步:

  1. 从后往前遍历找到第一个顺序的值下标为i,如[1,4,2,6,3,1]中为2和6,此时i为2
  2. 从后往前找到第一个比nums[i]大的数下标为j,如[1,4,2,6,3,1]中为3
  3. 交换i和j的值,然后反转i之后的数组
  • 时间复杂度:O(n),n为数组长度,最多遍历两遍
  • 空间复杂度:O(1)
class Solution {
    public void nextPermutation(int[] nums) {
        int i = nums.length - 2;
        while (i >= 0 && nums[i] >= nums[i + 1]) {
            i--;
        }
        if (i >= 0) {
            int j = nums.length - 1;
            while (j >= 0 && nums[i] >= nums[j]) {
                j--;
            }
            swap(nums, i, j);
        }
        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 left = start, right = nums.length - 1;
        while (left < right) {
            swap(nums, left, right);
            left++;
            right--;
        }
    }
}
posted @ 2021-07-19 16:27  你也要来一颗长颈鹿吗  阅读(24)  评论(0)    收藏  举报