当前位置: 代码迷 >> 综合 >> leetcode 随笔 Permutations,Permutations II
  详细解决方案

leetcode 随笔 Permutations,Permutations II

热度:90   发布时间:2024-01-04 09:09:09.0

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;}
};