Given a collection of distinct integers, return all possible permutations.
Example:
Input: [1,2,3] Output: [ [1,2,3], [1,3,2], [2,1,3], [2,3,1], [3,1,2], [3,2,1] ]
解法1
解题思路:
- 使用递归来解,用一个数组保存当前排列中的数字是否出现过,代码如下
class Solution {
public:
vector<vector<int>> permute(vector<int>& nums) {
int n = nums.size();
vector<vector<int> > result;
vector<int> path; // 临时保存当前的排列元素
vector<bool> visited(n, false); // 标记当前元素是否已经在临时排列中
permute(nums, 0, result, path, visited);
return result;
}
void permute(vector<int>& nums, int start, vector<vector<int> >& result, vector<int>& path, vector<bool>& visited) {
if(start == nums.size()) {
result.push_back(path);
return;
}
for(int i = 0; i < nums.size(); i++) {
if(visited[i]) continue;
path.push_back(nums[i]);
visited[i] = true;
permute(nums, start + 1, result, path, visited);
path.pop_back();
visited[i] = false;
}
}
};
解法2
解题思路:
- 还是使用递归来求解,只不过不需要visited数组来保存,直接从临时保存当前的排列中去查找元素是否存在,代码如下:
class Solution {
public:
vector<vector<int>> permute(vector<int>& nums) {
int n = nums.size();
vector<vector<int> > result;
vector<int> path; // 临时保存当前的排列元素
permute(nums, result, path);
return result;
}
void permute(vector<int>& nums, vector<vector<int> >& result, vector<int>& path) {
if(path.size() == nums.size()) {
result.push_back(path);
return;
}
for(int i = 0; i < nums.size(); i++) {
// 如果当前元素在当前排列中已经存在,寻找下一个元素
if(find(path.begin(), path.end(), nums[i]) != path.end())
continue;
path.push_back(nums[i]);
permute(nums, result, path);
path.pop_back();
}
}
};
参考资料: