题目如下:
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].
分析如下:
两种思路。第一种,逐步插入生成。第二种,通过交换生成。以[1,2,3]为例来进行说明。
第一种:
第一步,生成[1],
第二步,把2,插入上一步的[1]的所有可能位置中,得到[2,1],[1,2],
第三步,把3,插入上一步的结果中。对于[2,1],3能插入的所有可能位置是3个,得到[3,2,1], [2,3,1],[2,1,3]。对于[1,2],3能插入的所有可能位置是3个,得到[3,1,2], [1,3,2], [1,2,3]。
如果输入数组有N个,就将上面的迭代进行N次。
上面的代码可以用递归或者迭代来实现。
第二种:
交换的思路。主要参考了这篇文章 和这篇文章。如下图所示。适合递归来实现。
我的代码:
//1: 思路一的迭代版
class Solution {
public:
vector<vector<int> > permute(vector<int> &num) {
vector<vector<int> > empty_vec_vec;
if(num.size()==0) return empty_vec_vec;
vector<int> inner_result(1, num[0]);
vector<vector<int> > outer_result(1, inner_result);
vector<vector<int> > new_outer_result;
for (int i = 1; i < num.size(); ++i ) {
for (int j = 0; j < outer_result.size(); ++j) {
int inner_size = outer_result[j].size();
for (int k = 0; k <= inner_size; ++k) {
inner_result = outer_result[j];
inner_result.insert(inner_result.begin() + k, num[i]);
//node insert(argv1, argv2), argv1 is the iterator position.
new_outer_result.push_back(inner_result);
}
}
outer_result = new_outer_result;
new_outer_result.clear();
}
return outer_result;
}
};
//2: 思路一的递归版
class Solution {
public:
vector<vector<int> > permute(vector<int> &num) {
if (num.size() == 1){
vector<int> inner_result(1,num[0]);
vector<vector<int> > outer_result(1,inner_result);
return outer_result;
} else {
vector<int> sub_num(num.begin() + 1, num.end());
vector<vector<int> > outer_result = permute(sub_num);
vector<vector<int> > new_outer;
vector<int> new_inner;
for (int i = 0; i < outer_result.size(); ++i) {
for (int j = 0; j <= outer_result[i].size(); ++j) {
new_inner = outer_result[i];
new_inner.insert(new_inner.begin() + j, num[0]);
new_outer.push_back(new_inner);
}
}
return new_outer;
}
}
};
//:3 思路二的递归版
class Solution {
private:
void swap(vector<int> &num, int i, int j) {
int temp = num[i];
num[i] = num[j];
num[j] = temp;
}
void dfs(vector<int> & num, int k, int n, vector<vector<int> > & result) {
if (k == n){
result.push_back(num);
}else {
for (int i = k; i <= n; ++i) {
swap(num, k, i);
dfs(num, k+1, n, result);
swap(num, k, i);//每次回到基准点如{1,2,3},然后进行for循环的下一步。
}
}
}
public:
vector<vector<int> > permute(vector<int> &num) {
vector<vector<int> > result;
dfs(num, 0, num.size() -1 , result);
return result;
}
};
参考资料: