leetcode 40.组合总和2 (dfs + 减枝)

class Solution {
public:
    vector<int> tmp;
    vector<vector<int>> res;
    vector<pair<int, int>> freq;
    int sz;
    void dfs(int target, int pos, int sum){
        if(sum == target){
            res.push_back(tmp);
            return;
        }
        //到末尾还没到target或者后续的freq已经大于剩余
        if(pos == freq.size() || freq[pos].first > target - sum) return;
        //不取该位置
        dfs(target, pos + 1, sum);
        for(int i = 1; i <= freq[pos].second; i++){
            tmp.push_back(freq[pos].first);
            dfs(target, pos + 1, sum + i * freq[pos].first);
        }
        //把当前pos的freq.first都弹出
        for(int i = 1; i <= freq[pos].second; ++i) {
            tmp.pop_back();
        }
    }
    //为了使找到的组合数都不重复,将candidates排序,然后用vector<pair>记录每个元素出现
    //的频次n,在深搜vector,每个位置搜索1....n次,这个就能保证最后的组合数不重复
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sz = candidates.size();
        sort(candidates.begin(), candidates.end());
        for(int i = 0; i < sz; i++){
            if(freq.empty() || candidates[i] != freq.back().first){
                freq.push_back(pair<int, int>(candidates[i], 1));
            }
            else{
                freq.back().second++;
            }
        }
        dfs(target, 0, 0);
        return res;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值