1. 程式人生 > 實用技巧 >leetcode-----40. 組合總和 II

leetcode-----40. 組合總和 II

程式碼(dfs)

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

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

    void dfs(vector<int> &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();
        }
    }
};