欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

(LeetCode 216)组合总和 III [DFS:简单入门 + 去重]

程序员文章站 2022-05-20 20:23:00
...

216. 组合总和 III
找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明:

所有数字都是正整数。
解集不能包含重复的组合。
示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

分析:
此题是下面两天的衍生(其实最简单)。
“组合总和”:https://blog.csdn.net/STILLxjy/article/details/83627703
“组合总和 II”:https://blog.csdn.net/STILLxjy/article/details/83660387

在经过以上两题之后,这题就非常简单了。
1:自己手动构造一个candidates[1,2,3,4,5,6,7,8,9]
2:使用int size 记录当前组合的大小。

AC代码:

class Solution {
public:
    vector<int> candidates = {1,2,3,4,5,6,7,8,9};
    
    void dfs(vector<int> cur, int sum, int size, int used, vector<vector<int>>& ans, int k, int target)
    {
        if(sum > target) return;
        if(size > k) return;
        if(sum == target && size == k)
        {
            ans.push_back(cur);
            return;
        }
        
        for(int i=used;i<9;i++)
        {
            vector<int> t = cur;
            t.push_back(candidates[i]);
            int s = t.size();
            dfs(t,sum+candidates[i],s,i+1,ans,k,target);
            if(sum+candidates[i] > target) break;
        }
    }
    
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<vector<int>> ans;
        if(k == 0) return ans;
        vector<int> cur;
        dfs(cur,0,0,0,ans,k,n);
        return ans;
    }
};
相关标签: DFS