1. 程式人生 > 其它 >[題解]劍指 Offer 38. 字串的排列(C++)

[題解]劍指 Offer 38. 字串的排列(C++)

題目

輸入一個字串,打印出該字串中字元的所有排列。

你可以以任意順序返回這個字串陣列,但裡面不能有重複元素。

示例:

輸入:s = "abc"
輸出:["abc","acb","bac","bca","cab","cba"]

限制:

1 <= s 的長度 <= 8

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/zi-fu-chuan-de-pai-lie-lcof
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

思路

這題可以用回溯,但是我第一反應是用C++的標準庫函式next_permutation()。當然如果面試直接用輪子可能就到點了,所以還是自己實現一下,思路和庫函式的實現一致,可以看一下

https://en.cppreference.com/w/cpp/algorithm/next_permutation
時間複雜度O(n×n!),空間複雜度O(1)。

程式碼

class Solution {
public:
    vector<string> permutation(string s) {
        vector<string> ans;
        sort(s.begin(), s.end());
        do{
            ans.push_back(s);
        }while(nextPermutation(s));
        return ans;
    }

    bool nextPermutation(string& s){
        int n = s.size();
        if(n <= 1) return false;
        int i = n - 2;
        while(i >= 0 && s[i] >= s[i + 1]){
            --i;
        }
        if(i < 0){
            return false;
        }
        else{
            int j = n - 1;
            while(j > i && s[j] <= s[i]){
                --j;
            }
            swap(s[i], s[j]);
            reverse(s.begin() + i + 1, s.end());
        }
        return true;
    }
};