LeetCode 40. 组合总和 II(dfs)

问题描述

给定一个数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用一次。

说明：

所有数字（包括目标数）都是正整数。
解集不能包含重复的组合。
示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]

思路

可以先排下序让相同的数排在一起，然后开始搜索，在深搜中判断下相同的数有几个(这一题可能有重复元素但每个数只可以选一次)，接下来的循环判断每个数可以选几个再加上上面判断的条件就可以了

AC代码

class Solution {
public:

    vector<vector<int>> ans;
    vector<int> path;

    void solve(vector<int>& c, int u, int target) {
        if (target == 0) {
            ans.push_back(path);
            return;
        }

        if (u == c.size()) return;

        int k = u + 1;
        while (k < c.size() && c[k] == c[u]) k ++;
        int cnt = k - u;

        for (int i = 0; c[u] * i <= target && i <= cnt; i ++) {
            solve(c, k, target - c[u] * i);
            path.push_back(c[u]);
        }

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

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