LeetCode第 39 题:组合总数(C++)

39. 组合总和 - 力扣(LeetCode)

并没有说数组是否是有序的,又因为元素都是正整数,所以肯定要先排序啦。排序之后肯定只在比target小的区间里面进行选取。

注意元素可以重复选取

那似乎就需要从最小的数开始枚举了,所以应该使用回溯法,剪枝就很简单,当前和大于target就剪掉呗。ps, 用加法减法都可以。

那就直接上回溯思路,一次过:

class Solution {
public:
    vector<vector<int>> res; 
    vector<int> path;
    //idx为当前索引,当target == 0的时候符合要求
    void backtrack(vector<int> &a, int idx, int target){
        if(target == 0){
            res.push_back(path);
            return;
        }
        for(int i = idx; i < a.size(); ++i){//因为可以重复选取,所以这儿 i 得从传入的idx开始
            if(target - a[i] < 0)   break;
            path.push_back(a[i]);//做出选择
            backtrack(a, i, target - a[i]);
            path.pop_back();//撤销选择
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        sort(candidates.begin(), candidates.end());
        backtrack(candidates, 0, target);
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_32523711/article/details/107863364