leetcode 40. 组合总和 II

最新推荐文章于 2024-05-21 23:29:47 发布

原创最新推荐文章于 2024-05-21 23:29:47 发布 · 158 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

算法学习专栏收录该内容

30 篇文章

订阅专栏

本文探讨了如何使用深度优先搜索(DFS)解决组合总和问题，特别是当存在重复元素时的处理策略。通过先对输入数组排序，然后在递归过程中避免重复选择相同元素，有效地解决了重复解的问题。

这两天偷懒了，没有写题解=.=

dfs 模板题

虽说是模板题，还是要理解具体思想。因为题目中说到，每个元素不能重复利用，并且 candidates 可能有重复元素，所以我们要先把 candidates 进行排序。接下来就是深搜的过程了，由于每次搜索都是建立在上次搜索的结果之上，如果candidates[i] + sum <= target 那么就进行下一次搜索，并且把 candidates[i] 放入当前结果集中去，否则剪枝。还有一点就是要解决重复问题，这个问题在我们进行过排序之后就可以很容易的到解决，只要判断 i > start && candidates[i] == candidates[i-1] 就可以了，如果相等，那么就跳过此次循环。

class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<int> flag;
        vector<vector<int>> ans;

        sort(candidates.begin(), candidates.end());
        dfs(ans, candidates, flag, 0, target, 0);

        return ans;
    }

    void dfs(vector<vector<int>>& ans, vector<int>& candidates,vector<int>& flag,
        int sum, int target, int start){
        
        if(target == sum){
            ans.push_back(flag);
            return ;
        }else{
            for(int i = start; i < candidates.size(); i++){
                if(candidates[i] + sum <= target){
                    if(i != start && candidates[i] == candidates[i-1]){
                        continue;
                    }

                    flag.push_back(candidates[i]);
                    dfs(ans, candidates, flag, sum + candidates[i], target, i+1);
                    flag.pop_back();
                }else{
                    break;
                }
            }
        }
    }
};