分类 深度优先搜索 下的文章

子集 II

问题描述

给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。

示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subsets-ii

思路

基本思路和子集一样 https://www.irene.ink/search/%E5%AD%90%E9%9B%86/,只是最后删除重复的元素即可。
比子集多了,三个排序,一个去重。

  • nums进行排序
  • tmp进行排序后再插入到ret里面
  • ret进行排序和去重

题解

class Solution {
public:
    vector<int> tmp;
    vector<vector<int>> ret;
    int n;
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        n=nums.size();
        if(n==0) return ret;
        sort(nums.begin(),nums.end());
        ret.push_back({});
        dfs(nums,0,0);
        sort(ret.begin(),ret.end());
        ret.erase(unique(ret.begin(),ret.end()),ret.end());
        return ret;
    }
    void dfs(vector<int> &nums,int dep,int location){
        if(tmp.size()==n){
            return;
        }
        for(int i=location;i<n;i++){
            tmp.push_back(nums[i]);
            sort(tmp.begin(),tmp.end());
            ret.push_back(tmp);
            dfs(nums,dep+1,i+1);
            tmp.pop_back();

        }
    }
};

子集

问题描述

给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。

说明:解集不能包含重复的子集。

示例:
输入: nums = [1,2,3]
输出:
[
[3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subsets

思路

最后得到的结果是vector<vector> ret,在存进ret之前要先存入vector tmp;
因为nums的子集存在空集,所以,必须先在ret中存一个空的tmp;
然后,用深度优先,存入ret的顺序为:[1],[1,2],[1,2,3],到这里深度已经到底(因为每个元素只能用一次),返回到dfs的上一层,并将3弹出,3已经到达n-1即最后一个位置,所以再次返回到dfs的上一层,接下来存入[1,3],[2],[2,3],[3]

题解

class Solution {
public:
    vector<int> tmp;
    vector<vector<int>> ret;
    int n;
    int vis[100]={false};
    vector<vector<int>> subsets(vector<int>& nums) {
        n=nums.size();
        ret.push_back(tmp);
        dfs(nums,0);
        return ret;
    }
    void dfs(vector<int> &nums,int location){
        if(tmp.size()==n){
            return;
        }
        for(int i=location;i<n;i++){
            if(vis[i]== false){
                vis[i]=true;
                tmp.push_back(nums[i]);
                ret.push_back(tmp);
                dfs(nums,i+1);
                tmp.pop_back();
                vis[i]= false;
            }
        }
    }
};