给定一个不含重复数字的数组 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 中的所有整数 互不相同
回溯法,每次选择一个数填入,可以使用标记数组来记录已经填入的数
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
| class Solution { public: unordered_map<int, bool> vis; vector<vector<int>> ans; vector<vector<int>> permute(vector<int> &nums) { vector<int> mid; per(nums, nums.size(), 0, mid); return ans; } void per(vector<int> &nums, int n, int len, vector<int> &mid) { if (n == len) { ans.push_back(mid); return; } for (int i : nums) { if (vis[i]) continue; vis[i] = true; mid.push_back(i); per(nums, n, len + 1, mid); mid.pop_back(); vis[i] = false; } } };
|
因为题目没要求按字典序,所以可以直接把要填入的数交换到前面,这样nums就可以分成两段
前面一段表示已经确定,后面一段表示还未确定
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
| class Solution { public: vector<vector<int>> ans; vector<vector<int>> permute(vector<int> &nums) { per(nums, nums.size(), 0); return ans; } void per(vector<int> &nums, int n, int len) { if (n == len) { ans.push_back(nums); return; } for (int i = len; i < n; i++) { swap(nums[i], nums[len]); per(nums, n, len + 1); swap(nums[i], nums[len]); } } };
|