1. 程式人生 > >[LeetCode] Permutations 全排列

[LeetCode] Permutations 全排列

Given a collection of numbers, return all possible permutations.

For example,
[1,2,3] have the following permutations:
[1,2,3][1,3,2][2,1,3][2,3,1][3,1,2], and [3,2,1].

這道題是求全排列問題,給的輸入陣列沒有重複項,這跟之前的那道 Combinations 組合項 和類似,解法基本相同,但是不同點在於那道不同的數字順序只算一種,是一道典型的組合題,而此題是求全排列問題,還是用遞迴DFS來求解。這裡我們需要用到一個visited陣列來標記某個數字是否訪問過,然後在DFS遞迴函式從的迴圈應從頭開始,而不是從level開始,這是和

 Combinations 組合項 不同的地方,其餘思路大體相同,程式碼如下:

解法一

class Solution {
public:
    vector<vector<int> > permute(vector<int> &num) {
        vector<vector<int> > res;
        vector<int> out;
        vector<int> visited(num.size(), 0);
        permuteDFS(num, 
0, visited, out, res); return res; } void permuteDFS(vector<int> &num, int level, vector<int> &visited, vector<int> &out, vector<vector<int> > &res) { if (level == num.size()) res.push_back(out); else { for
(int i = 0; i < num.size(); ++i) { if (visited[i] == 0) { visited[i] = 1; out.push_back(num[i]); permuteDFS(num, level + 1, visited, out, res); out.pop_back(); visited[i] = 0; } } } } };

還有一種遞迴的寫法,更簡單一些,這裡是每次交換num裡面的兩個數字,經過遞迴可以生成所有的排列情況,程式碼如下:

解法二

class Solution {
public:
    vector<vector<int> > permute(vector<int> &num) {
        vector<vector<int> > res;
        permuteDFS(num, 0, res);
        return res;
    }
    void permuteDFS(vector<int> &num, int start, vector<vector<int> > &res) {
        if (start >= num.size()) res.push_back(num);
        for (int i = start; i < num.size(); ++i) {
            swap(num[start], num[i]);
            permuteDFS(num, start + 1, res);
            swap(num[start], num[i]);
        }
    }
};

最後再來看一種方法,這種方法是CareerCup書上的方法,也挺不錯的,這道題是思想是這樣的:

當n=1時,陣列中只有一個數a1,其全排列只有一種,即為a1

當n=2時,陣列中此時有a1a2,其全排列有兩種,a1a2和a2a1,那麼此時我們考慮和上面那種情況的關係,我們發現,其實就是在a1的前後兩個位置分別加入了a2

當n=3時,陣列中有a1a2a3,此時全排列有六種,分別為a1a2a3, a1a3a2, a2a1a3, a2a3a1, a3a1a2, 和 a3a2a1。那麼根據上面的結論,實際上是在a1a2和a2a1的基礎上在不同的位置上加入a3而得到的。

_ a_ a_ : a3a1a2, a1a3a2, a1a2a3

_ a_ a_ : a3a2a1, a2a3a1, a2a1a3

解法三:

class Solution {
public:
    vector<vector<int> > permute(vector<int> &num) {
        if (num.empty()) return vector<vector<int> >(1, vector<int>());
        vector<vector<int> > res;
        int first = num[0];
        num.erase(num.begin());
        vector<vector<int> > words = permute(num);
        for (auto &a : words) {
            for (int i = 0; i <= a.size(); ++i) {
                a.insert(a.begin() + i, first);
                res.push_back(a);
                a.erase(a.begin() + i);
            }
        }   
        return res;
    }
};

類似題目: