leetcode解题: Permutations (46)

发表于 | 分类于

Given a collection of distinct numbers, return all possible permutations.

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

1
2
3
4
5
6
7
8
[
  [1,2,3],
  [1,3,2],
  [2,1,3],
  [2,3,1],
  [3,1,2],
  [3,2,1]
]

解法1： O(n!)

也是经典的backtracking的问题，对于这种permutation的问题， 要维护一个visited数组来记录每一个元素是否被选取了，因为这里每次都是从头开始扫描。 递归终止的条件是当选取的答案的长度和原数组的长度一致的时候就说明一个permutation已经完成。
因为有n!个组合，每一个组合访问一次，总的复杂度是O（n!).
C++

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
class Solution {
public:
    vector<vector<int>> permute(vector<int>& nums) {
        vector<vector<int>> res;
        if (nums.empty()) {
            return res;
        }
        
        vector<bool> visited (nums.size(), false);
        vector<int> cur;
        helper(nums, visited, cur, res);
        return res;
    }
    
    void helper(vector<int>& nums, vector<bool>& visited, vector<int>& cur, vector<vector<int>>& res) {
        
        if (cur.size() == nums.size()) {
            res.push_back(cur);
            return;
        }
        
        for (int i = 0; i < nums.size(); ++i) {
            if (!visited[i]) {
                cur.push_back(nums[i]);
                visited[i] = true;
                helper(nums, visited, cur, res);
                cur.pop_back();
                visited[i] = false;
            }
        }
        return;
        
    }
};

Java

1