LeetCode216. 组合总和 III

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Jaster_wisdom/article/details/81904572

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

说明:

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

示例 1:

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

示例 2:

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

题目分析:DFS,为避免重复组合,规定数字只能往后取。例如,当前取了数字2,那么下一个数只能取2之后的数字。

代码展示:

class Solution {
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<int> path;
        vector<vector<int>> ans;
        int depth = 0;
        int sum = 0;
        int index = 1;
        dfs(k,n,index,sum,depth,path,ans);
        return ans;
    }
    
    void dfs(int k,int n,int index,int sum,int depth,vector<int>& path,vector<vector<int>>& ans){
        if(depth==k && sum==n){
            ans.push_back(path);
            return;
        }
        if(depth>=k)
            return;
        for(int i=index;i<10;i++){
            if(sum+i<=n){
                path.push_back(i);
                dfs(k,n,i+1,sum+i,depth+1,path,ans);
                path.pop_back();    //回溯
            }
        }
    }
};

 

没有更多推荐了,返回首页