题目:
好久没有刷题了,今天遇到了板子题都还想了一会,秋招是不是可以预先说GG了。hhhh
这种类型题目的通解就是DFS,某种程度上来说也是暴力的一种,这里附上板子,基本上所有的DFS的类型都可以按照这个板子来做,包括矩阵(可能涉及到一个方向选择的问题)
vector<int> t;
void dfs(int cur, int n) {
if (cur == n) {
// 记录答案
// ...
return;
}
// 考虑选择当前位置
t.push_back(cur);
dfs(cur + 1, n, k);
t.pop_back();
// 考虑不选择当前位置
dfs(cur + 1, n, k);
}
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/subsets/solution/zi-ji-by-leetcode-solution/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
那么回到我们这道题,按照板子来吧。
第一部分是返回条件,那么我们很容易想到只要当前位置超过了整个数组位置就直接返回。同时需要记录这一次遍历的结果,并且将它推入答案数组中。
注意,这里不论是放在if循环里面还是外面都可以,只有一个空子集需要额外处理的问题。
第二部分是当前位置该做的事情,这一题就只需要简单的将元素推入数组即可。
第三部分是往下进一步搜索,也就是继续往下DFS遍历
第四部分是还原变化,即将队尾的元素推出,然后进一步遍历
最后返回即可
class Solution {
public:
vector<int> sup;
vector<vector<int>> ans;
void dfs(int i,vector<int>& nums){
if(i>nums.size()-1){
ans.emplace_back(sup);
return;
}
sup.emplace_back(nums[i]);
dfs(i+1,nums);
sup.pop_back();
dfs(i+1,nums);
}
vector<vector<int>> subsets(vector<int>& nums) {
dfs(0,nums);
return ans;
}
};
放在if外面,即从短的开始记录,上面是从最长的开始记录,注意要额外推入一个空子集,并且推入顺序不能反
class Solution {
public:
vector<int> sup;
vector<vector<int>> ans;
void dfs(int i,vector<int>& nums){
if(i>nums.size()-1){
return;
}
sup.emplace_back(nums[i]);
ans.emplace_back(sup);
dfs(i+1,nums);
sup.pop_back();
dfs(i+1,nums);
}
vector<vector<int>> subsets(vector<int>& nums) {
ans.push_back(sup);
dfs(0,nums);
return ans;
}
};