leetcode-----40. 组合总和 II

  • 链接:https://leetcode-cn.com/problems/combination-sum-ii/

代码(dfs)

class Solution {
public:
    vector> ans;
    vector path;

    vector> combinationSum2(vector& candidates, int target) {
        sort(candidates.begin(), candidates.end());
        dfs(candidates, 0, target);
        return ans;    
    }

    void dfs(vector &cs, int u, int target) {
        if (target == 0) {
            ans.push_back(path);
            return;
        }
        if (u == cs.size()) return;

        int k = u + 1;
        while (k < cs.size() && cs[k] == cs[u]) k++;
        int cnt = k - u;
        
        for (int i = 0; cs[u] * i <= target && i <= cnt; ++i) {
            dfs(cs, k, target - cs[u] * i);
            path.push_back(cs[u]);    
        }

        for (int i = 0; cs[u] * i <= target && i <= cnt; ++i) {
            path.pop_back();
        }
    }
};

你可能感兴趣的:(leetcode-----40. 组合总和 II)