leetcode78.子集➕90.子集2

78子集

dfs

dfs1:
和全排列的区别就是对于当前考察的索引i,全排列如果不取i,之后还要取i,所以需要一个visited数组用来记录。对于子集问题如果不取i,之后也不必再取i。

单纯递归回溯

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<int> cur;
        vector<vector<int>>res;
        dfs(res,cur,nums,0);
        return res;
    }
    void dfs(vector<vector<int>>&res,vector<int>&cur,vector<int>&nums,int nex){
        res.push_back(cur);
        for(int i=nex;i<nums.size();++i){
            cur.push_back(nums[i]);
            dfs(res,cur,nums,i+1);
            cur.pop_back();
        }
    }
};

dfs2:类似树的递归遍历,一个dfs函数里有两个dfs递归子函数,一个是含当前数字的子递归,一个是不含当前数字的子递归。

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<int>cur;
        vector<vector<int>>res;
        dfs(res,cur,nums,0);
        return res;
    }
    void dfs(vector<vector<int>>&res,vector<int>&cur,vector<int>&nums,int nex){
        if(nex>=nums.size()){
            res.push_back(cur);
            return;
        }
        cur.push_back(nums[nex]);
        dfs(res,cur,nums,nex+1);
        cur.pop_back();
        dfs(res,cur,nums,nex+1);
    }
};

位运算

因为所以数字都不一一样,每个数字都有两种可能,一个取一个不取。假设一共n个元素,则一共有2的n次方个子集,其中每个子集都对应一个小于2的n次方的数。如n==4,1001对应首尾数字取,中间两个数字不取

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        int n=nums.size();
        vector<vector<int>>res;
        for(int i=0;i<pow(2,n);++i){
            int j=i,k=n-1;
            res.push_back({});
            while(j){
                if(j&1){
                    res.back().push_back(nums[k]);
                }
                j>>=1;
                --k;
            }
        }
        return res;
    }
};

循环

空生空+【1】,空加【1】生空+【1】加【2】加【1,2】。
即1生2,2生4,4生8。。。。

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>>res={{}};
        for(const int& num:nums){
            int cur_size=res.size();
            for(int i=0;i<cur_size;++i){
                res.push_back(res[i]);
                res.back().push_back(num);
            }
        }
        return res;
    }
};

90 子集2

循环

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<vector<int>>res={{}};
        unordered_map<int,int> mp;
        for(int num:nums){
            mp[num]+=1;
        }
        for(const auto& pai:mp){
            int cur_size=res.size();
            for(int i=0;i<cur_size;++i){
                res.push_back(res[i]);
                for(int j=0;j<pai.second;++j){
                    res.back().push_back(pai.first);
                    res.push_back(res.back());
                }
                res.pop_back();
            }
        }
        return res;
    }
};

dfs

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        vector<int>cur;
        dfs(res,nums,cur,0);
        return res;
    }
    void dfs(vector<vector<int>>&res,vector<int>&nums,vector<int>&cur,int nex){
        res.push_back(cur);
        for(int i=nex;i<nums.size();++i){
            while(i-1>=nex and i<nums.size() and nums[i]==nums[i-1]){
                ++i;
            }
            if(i>=nums.size()){return;}
            cur.push_back(nums[i]);
            dfs(res,nums,cur,i+1);
            cur.pop_back();
        }
    }
};
题目描述: 给定一个只包含正整数的非空数组,是否可以将这个数组分成两个子集,使得两个子集的元素和相等。 示例: 输入:[1, 5, 11, 5] 输出:true 解释:数组可以分割成 [1, 5, 5] 和 [11]。 解题思路: 这是一道经典的 0-1 背包问题,可以使用动态规划或者回溯算法解决。 使用回溯算法,需要定义一个 backtrack 函数,该函数有三个参数: - 数组 nums; - 当前处理到的数组下标 index; - 当前已经选择的元素和 leftSum。 回溯过程中,如果 leftSum 等于数组元素和的一半,那么就可以直接返回 true。如果 leftSum 大于数组元素和的一半,那么就可以直接返回 false。如果 index 到达数组末尾,那么就可以直接返回 false。 否则,就对于当前元素,有选择和不选择两种情况。如果选择当前元素,那么 leftSum 就加上当前元素的值,index 就加 1。如果不选择当前元素,那么 leftSum 不变,index 也加 1。最终返回所有可能性的结果中是否有 true。 Java 代码实现: class Solution { public boolean canPartition(int[] nums) { int sum = 0; for (int num : nums) { sum += num; } if (sum % 2 != 0) { return false; } Arrays.sort(nums); return backtrack(nums, nums.length - 1, sum / 2); } private boolean backtrack(int[] nums, int index, int leftSum) { if (leftSum == 0) { return true; } if (leftSum < 0 || index < 0 || leftSum < nums[index]) { return false; } return backtrack(nums, index - 1, leftSum - nums[index]) || backtrack(nums, index - 1, leftSum); } }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值