LeetCode题解——40. 组合总和 II

题目相关

题目链接

LeetCode中国，https://leetcode-cn.com/problems/combination-sum-ii/。

题目描述

给定一个数组 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]
]

题目分析

LeetCode 给出本题难度中等。

本题是 LeetCode 编号 39 题目的升级版，这个题目的题解可以参考，https://blog.youkuaiyun.com/justidle/article/details/105316988。只需要增加剪枝操作即可。

细节

解集不能包含重复的组合

我们可以通过排序来解决。

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

和题号 39 相比，下一次搜索为本次搜索位置加一。

AC 参考代码

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, 0, target);
        return ans;        
    }

    void dfs(vector<int>& candidates, int sum, int pos, int target) {
        if (sum==target) {
            ans.push_back(path);
            return;
        } else if (sum>target) {
            return;
        }

        int val;
        for (int i=pos; i<candidates.size(); i++) {
            //减枝操作
            if (i>pos && candidates[i]==candidates[i-1]) {
                continue;
            }
            val = candidates[i];
            if (sum+val<=target) {
                path.push_back(val);
                dfs(candidates, sum+val, i+1, target);
                path.pop_back();
            }
        }
    }    
};