LeetCode 31 Next Permutation

題意:

給出一串數字,求該排列的下一個排列。如果該排列爲字典序最大排列,則輸出字典序最小排列。


思路:

首先C++裏面有求下一個排列的函數next_permutation,該函數返回0即表示當前排列字典序最大。

如果要自己動手實現,那麼就考慮“如何找到比當前排列大1的排列”。

假設排列A(aaaXddd)比排列B(aaaYfff)大1,a部分爲兩排列相同部分,X與Y表示最靠左邊不同的那一位。那麼,X>Y且X一定在B中的f部分且f中數字一定是遞減排序的且d一定是遞增排序的。


代碼:

class Solution {
public:
    void nextPermutation(vector<int> &nums) {
        int idx = findPos(nums);
        if (idx == -1) {
            reverse(nums.begin(), nums.end());
        } else {
            reverse(nums.begin() + idx + 1, nums.end());
            for (int i = idx + 1; i < nums.size(); ++i) {
                if (nums[idx] < nums[i]) {
                    swap(nums[idx], nums[i]);
                    break;
                }
            }
        }
    }

private:
    int findPos(vector<int> &nums) {
        for (int i = nums.size() - 2; i >= 0; --i) {
            if (nums[i] < nums[i + 1]) {
                return i;
            }
        }
        return -1;
    }
};

/**
 * 偷懶版本……
 */

class Solution {
public:
    void nextPermutation(vector<int> &nums) {
        if (!next_permutation(nums.begin(), nums.end())) {
            sort(nums.begin(), nums.end());
        }
    }
};


發佈了285 篇原創文章 · 獲贊 41 · 訪問量 22萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章