[C++]LeetCode40.组合总和 II

最新推荐文章于 2024-01-16 15:15:19 发布

原创最新推荐文章于 2024-01-16 15:15:19 发布 · 373 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#c++ #数据结构 #c++

Leetcode刷题专栏收录该内容

133 篇文章

订阅专栏

这篇博客介绍了如何利用深度优先搜索（DFS）解决寻找数组中数字组合之和为目标数的问题。提供了两种不同的DFS实现，分别通过set和vector来避免重复组合，并通过排序优化搜索过程。示例展示了对于特定输入数组和目标数的解集。

40.组合总和 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]
]

解法一：
class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        //set可以判断是否有重复
        set<vector<int>> ans;
        vector<int> cur;
        sort(candidates.begin(), candidates.end());
        dfs(candidates, target, 0, ans, cur);
        //最终再把set转换为vector类型
        return vector<vector<int>>(ans.begin(),ans.end());
    }
    void dfs(vector<int>& candidates, int target, int s, set<vector<int>>& ans, vector<int>& cur)
    {
        //target为零时，说明找到了答案，则返回
        if(target == 0)
        {
            ans.insert(cur);
            return;
        }
        //for循环遍历数组长度（横向）
        for(int i = s; i < candidates.size(); i++)
        {
            //当前元素大于target时，一定不满足条件，退出循环
            if(candidates[i] > target)
                return;
            //存入当前元素
            cur.push_back(candidates[i]);
            //递归（纵向） 注意：因为可以重复使用元素，所以是i而不是i+1
            dfs(candidates, target-candidates[i], i+1, ans, cur);
            //撤销存入的元素，以便进行下一次回溯
            cur.pop_back();

        }
    }
};
解法二：
class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<vector<int>> ans;
        vector<int> cur;
        sort(candidates.begin(), candidates.end());
        dfs(candidates, target, 0, ans, cur);
        return ans;
    }
    void dfs(vector<int>& candidates, int target, int s, vector<vector<int>>& ans, vector<int>& cur)
    {
        //target为零时，说明找到了答案，则返回
        if(target == 0)
        {
            ans.push_back(cur);
            return;
        }
        //for循环遍历数组长度（横向）
        for(int i = s; i < candidates.size(); i++)
        {
            //当前元素大于target时，一定不满足条件，退出循环
            if(candidates[i] > target)
                return;
            if(i > s && candidates[i] == candidates[i-1])
                continue;
            //存入当前元素
            cur.push_back(candidates[i]);
            //递归（纵向） 注意：因为可以重复使用元素，所以是i而不是i+1
            dfs(candidates, target-candidates[i], i+1, ans, cur);
            //撤销存入的元素，以便进行下一次回溯
            cur.pop_back();

        }
    }
};