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] 有两个位置插入,插入2 变成
[1,2]和[2,1] 然后有三个位置插入 ,插入3 变成
[3,1,2] [1,3,2] [1,2,3] [3,2,1] [2,3,1] [2,1,3]
class Solution {
public:
vector<vector<int>> permute(vector<int>& nums) {
vector<vector<int>> res;
if (nums.empty()) return res;
vector<int> t; t.push_back(nums[0]);
res.push_back(t);
for (int i = 1; i < nums.size(); i++)
{
vector<vector<int>> tmp;
for (int j = 0; j < res.size(); j++)
{
vector<int> kk = res[j];
for (int t = 0; t <= i; t++)
{
kk.insert(kk.begin() + t, nums[i]);
tmp.push_back(kk);
kk.erase(kk.begin() + t);
}
}
res = tmp;
}
return res;
}
};
=================================================
=================================================
Given a collection of numbers that might contain duplicates, return all possible unique permutations.
Example:
Input: [1,1,2] Output: [ [1,1,2], [1,2,1], [2,1,1] ]
这次给的数组存在重复的数字了,求全排列。
最直接的解法就是上面的代码+去重。试了一下跑是跑过了,时间太长57ms
class Solution {
public:
vector<vector<int>> permuteUnique(vector<int>& nums) {
vector<vector<int>> res;
if (nums.empty()) return res;
vector<int> t; t.push_back(nums[0]);
res.push_back(t);
for (int i = 1; i < nums.size(); i++)
{
vector<vector<int>> tmp;
for (int j = 0; j < res.size(); j++)
{
vector<int> kk = res[j];
for (int t = 0; t <= i; t++)
{
if (t < i&&nums[i] != kk[t])
{
kk.insert(kk.begin() + t, nums[i]);
tmp.push_back(kk);
kk.erase(kk.begin() + t);
}
else if (t == i)
{
kk.insert(kk.begin() + t, nums[i]);
tmp.push_back(kk);
kk.erase(kk.begin() + t);
}
}
}
res = tmp;
for (int i = 0; i < res.size(); i++)
{
for (int k = i + 1; k < res.size(); k++)
{
if (res[i] == res[k])
res.erase(res.begin() + k);
}
}
}
return res;
}
};
看了一下discuss,发现大多数优秀的答案都是利用递归
递归的思路是先排序,然后轮流选取不同的数当 这串数字的 “头”
以[1,2,3]为例
首先确定“头”的位置,头的位置首先在 位置0
可以在“头”位置的可以是1,2,3
即[1,2,3] [2,1,3] [3,1,2]
然后"头"位置移动到下一位置,位置1
对于[1,2,3] “头”是 2 ,头可以是 2或3
所以就有最终答案[1,2,3] [1,3,2]
对于[2,1,3] “头”是 1 ,头可以是 1或3
所以就有最终答案[2,1,3] [2,3,1]
对于[3,1,2] “头”是 1 ,头可以是 1或2
所以就有最终答案[3,1,2] [3,2,1]
对于重复数字不进行"换头"的操作class Solution {
public:
void recursion(vector<int> nums, int i, vector<vector<int> > &res) {
if (i == nums.size()-1) {
res.push_back(nums);
return;
}
for (int k = i; k < nums.size(); k++) {
if (i != k && nums[i] == nums[k]) continue;
swap(nums[i], nums[k]);
recursion(nums, i+1, res);
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
sort(nums.begin(),nums.end());
vector<vector<int>> res;
recursion(nums,0,res);
return res;
}
};