leetcode 随笔 Permutations,Permutations II

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



阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页