31. 下一個排列

东岸發表於2024-05-23

整數陣列的一個 排列 就是將其所有成員以序列或線性順序排列。

例如,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]

class Solution {
public:
    void nextPermutation(vector<int>& nums) {
        int n=nums.size();
        int i=n-2;
        while(i>=0)
        {
            if(nums[i]<nums[i+1])
            {
                break;
            }
            i--;
        }
        if(i>=0)
        {
            int j=n-1;
            while(j>=0)
            {
                if(nums[j]>nums[i])
                {
                    break;
                }
                j--;
            }
            std::swap(nums[i],nums[j]);
        }
        std::reverse(nums.begin()+i+1,nums.end());

    }
};

總的來說,這個演算法的思路是:

從後向前找到第一個滿足 nums[i] < nums[i+1] 的位置 i。
在 nums[i+1:] 這個子陣列中找到一個比 nums[i] 大的最小值 nums[j],並交換 nums[i] 和 nums[j]。
反轉 nums[i+1:] 這個子陣列,得到下一個排列。

相關文章