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

Leetcode Combination Sum

Leetcode algorithms 第 39 题:Combination Sum。

题目

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

The same repeated number may be chosen from C unlimited number of times.

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 2,3,6,7 and target 7, 
A solution set is: 
[7] 
[2, 2, 3] 

解答

从数组中找出和为给定值的所有组合,每个数字可以使用无限次。使用dfs进行搜索,每次选择当前元素若干次,但总和不能超过target,最终到达sum时即终止本次递归。

具体代码如下:

class Solution {
public:
    void dfs(int x, int sum, vector<int> &cur, vector<int> &candidates, vector<vector<int>> &ans, int target)
    { //当前位置,当前和,当前每个数字已选择的次数,候选数字,答案,目标
        if (sum == target)
        {
            vector<int> tem;
            for (int i = 0; i < x; i++)
                for (int j = 0; j < cur[i]; j++)
                    tem.push_back(candidates[i]);
            ans.push_back(tem);
            return;
        }
        if (x == candidates.size()) return;
        
        int k = (target - sum) / candidates[x]; //最多选择次数
        for (int i = 0; i <= k; i++)
        {
            cur[x] = i;
            dfs(x+1, sum + candidates[x] * i, cur, candidates, ans, target);
        }
    }
    vector<vector<int> > combinationSum(vector<int> &candidates, int target) {
        sort(candidates.begin(), candidates.end());
        vector<int> cur(candidates.size(), 0);
        vector<vector<int>> ans;
        dfs(0, 0, cur, candidates, ans, target);
        return ans;
    }
};
上一篇: 下一篇:

我的博客

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

站内搜索

最新评论