Not Only Algorithm,不仅仅是算法,关注数学、算法、数据结构、程序员笔试面试以及一切涉及计算机编程之美的内容 。。
你的位置:NoAlGo博客 » 题解 » 

Leetcode Combination Sum II

Leetcode algorithms 第 40 题:Combination Sum II。

题目

Given a collection of candidate numbers (C) and a target number (T), find all unique combinations in C where the candidate numbers sums to T.

Each number in C may only be used once in the combination.

Note:

  • All numbers (including target) will be positive integers.
  • Elements in a combination (a1, a2, … , ak) must be in non-descending order. (ie, a1 ≤ a2 ≤ … ≤ ak).
  • The solution set must not contain duplicate combinations.

For example, given candidate set 10,1,2,7,6,1,5 and target 8, 
A solution set is: 
[1, 7] 
[1, 2, 5] 
[2, 6] 
[1, 1, 6] 

解答

从数组中找出和为给定值的所有组合,每个数字至多使用一次。

同样是使用dfs进行搜索,由于可能出现相同元素,先排序方便后续处理。dfs时每次处理一个数字,相同的数字一起处理,相同数字可以不选,也可以每次选择一个,但不能选择超过其出现次数的个数。

当和超过target时终止,而刚好为target时则找到一个答案。

具体代码如下:

class Solution {
public:
    void dfs(int id, int sum, vector<int> &vs, vector<int> &num, int target, vector<vector<int>> &ans)
    { //当前位置,当前和,该数字是否使用,候选数字,目标,答案
        if (sum >= target)
        {
            if (sum == target) //找到一个答案
            {
                vector<int> tem;
                for (int i = 0; i < num.size(); i++)
                    if (vs[i]) tem.push_back(num[i]);
                ans.push_back(tem);
            }
            return;
        }
        if (id == num.size()) return; //没得选了
        
        int j = id + 1; //下一个不相同的元素的位置
        while (j < num.size() && num[j] == num[id]) ++j;
        
        dfs(j, sum, vs, num, target, ans); //不选该数字
        
        for (int i = id; i < j; i++) //处理重复出现的数字
        {
           vs[i] = 1, sum += num[i]; //选择该数字一次,不会超过其出现的总次数
           dfs(j, sum, vs, num, target, ans);
        }
        for (int i = id; i < j; i++) //回溯
            vs[i] = 0, sum -= num[i];
    }
    vector<vector<int> > combinationSum2(vector<int> &num, int target) {
        sort(num.begin(), num.end());
        vector<int> vs(num.size(), 0);
        vector<vector<int> > ans;    
        dfs(0, 0, vs, num, target, ans);
        return ans;
    }
};
上一篇: 下一篇:

我的博客

NoAlGo头像编程这件小事牵扯到太多的知识,很容易知其然而不知其所以然,但真正了不起的程序员对自己程序的每一个字节都了如指掌,要立足基础理论,努力提升自我的专业修养。

站内搜索

最新评论