LeetCode 回溯算法 46. 全排列 C++, my write

难度中等

给定一个不含重复数字的数组 nums ，返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。

示例 1：

输入：nums = [1,2,3]
输出：[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

示例 2：

输入：nums = [0,1]
输出：[[0,1],[1,0]]

示例 3：

输入：nums = [1]
输出：[[1]]

提示：

1 <= nums.length <= 6
-10 <= nums[i] <= 10
nums 中的所有整数 互不相同

分析：使用回溯算法解决全排列问题，可以抽象为树形结构，每层循环选中一个元素往path数组加入元素，当path数组大小 == nums数组大小，说明遍历到最终叶子节点得到一个全排列结果，将path加入结果集res。创建一个新的bool类型数组used，初始化为false，若该元素被选中则设置为true，used数组来判断该元素是否可以加入path。

扫描二维码关注公众号，回复： 17450814 查看本文章

class Solution {
private:
	vector<int> path;	// 满足条件的一个结果 
	vector<vector<int>> res;	// 结果集 
	
	void backtracking(vector<int> nums, vector<bool> used){
		// 若path的个数和nums个数相等，说明得到一个结果 
		if(path.size() == nums.size()){
			res.push_back(path);
			return ;
		}
		
		for(int i = 0; i < nums.size(); i++){
			if(used[i] == true)	// 已经选过，跳过 
				continue;
			used[i] = true;	// 选择 
			path.push_back(nums[i]); // 加入元素 
			backtracking(nums, used); // 进入下一层循环 
			path.pop_back();	// 回溯, 去掉元素 
			used[i] = false; 	// 回溯，撤销选择 
		}
	}
public:
    vector<vector<int>> permute(vector<int>& nums) {
		
		path.clear();
		res.clear();
		vector<bool> used(nums.size(), false);
		
		backtracking(nums, used);
		
		return res;
    }
};

执行结果：

通过

执行用时：8 ms, 在所有 C++ 提交中击败了15.59%的用户

内存消耗：8.6 MB, 在所有 C++ 提交中击败了5.17%的用户

通过测试用例：26 / 26

LeetCode 回溯算法 46. 全排列 C++, my write

猜你喜欢