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].

http://oj.leetcode.com/problems/permutations/


Solution:

Recursion and dp, use a vector to store which digit has been visited.

https://github.com/starcroce/leetcode/blob/master/permuntations.cpp

// 68 ms for 25 test cases
// use a visited vector to store which digit has been visited
class Solution {
public:
    vector<vector<int> > permute(vector<int> &num) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        vector<vector<int> > res;
        if(num.size() == 0) {
            return res;
        }
        vector<int> sol;
        vector<bool> visited(num.size(), false);
        permute(num, 0, visited, res, sol);
        return res;
    }
    
    void permute(vector<int> &num, int level, vector<bool> visited, vector<vector<int> > &res, vector<int> sol) {
        if(sol.size() == num.size()) {
            res.push_back(sol);
            return;
        }
        for(int i = 0; i < num.size(); i++) {
            if(visited[i] == false) {
                visited[i] = true;
                sol.push_back(num[i]);
                permute(num, i+1, visited, res, sol);
                sol.pop_back();
                visited[i] = false;
            }
        }
    }
};


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章