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

  • Post author:
  • Post category:其他

46. 全排列

难度中等

给定一个不含重复数字的数组 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。

 

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


版权声明:本文为qq_43833393原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。