Given a collection of distinct 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].
解:
class Solution {
public:
vector< vector<int> > res;
vector<vector<int>> permute(vector<int>& nums) {
int length=nums.size();
if(length==0)return vector< vector<int> >();
if(length==1)return vector< vector<int> >{nums};
vector<int> prev;
permute(nums,prev);
return res;
}
void permute(vector<int>& nums,vector<int>& prev) //nums是还剩下的数字的数组,prev是已经排好的数组
{
int n=nums.size();
if(n==1){
prev.push_back(nums[0]);
res.push_back(prev);
}
for(int i=0;i<n;++i)
{
vector<int> tmp(nums);
vector<int> tmp_pre(prev);
tmp_pre.push_back(*(tmp.begin()+i));
tmp.erase(tmp.begin()+i);
permute(tmp,tmp_pre);
}
}
};
class Solution {
public:
vector<vector<int> > permute(vector<int> &num) {
vector<vector<int> > result;
permuteRecursive(num, 0, result);
return result;
}
// permute num[begin..end]
// invariant: num[0..begin-1] have been fixed/permuted
void permuteRecursive(vector<int> &num, int begin, vector<vector<int> > &result) {
if (begin >= num.size()) {
// one permutation instance
result.push_back(num);
return;
}
for (int i = begin; i < num.size(); i++) {
swap(num[begin], num[i]);
permuteRecursive(num, begin + 1, result);
// reset
swap(num[begin], num[i]);
}
}
};
还有一种
the basic idea is, to permute n numbers, we can add the nth number into the resulting List<List<Integer>> from the n-1 numbers, in every possible position.
For example, if the input num[] is {1,2,3}: First, add 1 into the initial List<List<Integer>> (let's call it "answer").
Then, 2 can be added in front or after 1. So we have to copy the List in answer (it's just {1}), add 2 in position 0 of {1}, then copy the original {1} again, and add 2 in position 1. Now we have an answer of {{2,1},{1,2}}. There are 2 lists in the current answer.
Then we have to add 3. first copy {2,1} and {1,2}, add 3 in position 0; then copy {2,1} and {1,2}, and add 3 into position 1, then do the same thing for position 3. Finally we have 2*3=6 lists in answer, which is what we want.
public List<List<Integer>> permute(int[] num) {
List<List<Integer>> ans = new ArrayList<List<Integer>>();
if (num.length ==0) return ans;
List<Integer> l0 = new ArrayList<Integer>();
l0.add(num[0]);
ans.add(l0);
for (int i = 1; i< num.length; ++i){
List<List<Integer>> new_ans = new ArrayList<List<Integer>>();
for (int j = 0; j<=i; ++j){
for (List<Integer> l : ans){
List<Integer> new_l = new ArrayList<Integer>(l);
new_l.add(j,num[i]);
new_ans.add(new_l);
}
}
ans = new_ans;
}
return ans;
}
还可以直接用 algorithm 标准库里面的next_permutation 函数
vector<vector<int> > permute(vector<int> &num) {
vector<vector<int> > ans;
sort(num.begin(), num.end());
ans.push_back(num);
while(next_permutation(num.begin(), num.end()))
ans.push_back(num);
return ans;
}