LeetCode 216.组合总和 Ⅲ(C++)

找出所有相加之和为 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]]

执行用时为0ms的

class Solution {
public:
    vector> combinationSum3(int k, int n) {
        vector> result;
        if(n < 3 || n > 45) return result;
        vector temp;
        for(int i = 1;i <= 9;i++) {
            dfs(result,temp,i,0,1,k,n);
        }
        return result;
    }
    void dfs(vector> & result,vector &temp,int cur,int sum,int index,int k,int n) {
        temp.push_back(cur);
        sum += cur;
        if(sum == n && index == k) {
            result.push_back(temp);
            temp.pop_back();
            sum -= cur;
            return;
        }
        else if(index == k) {
            temp.pop_back();
            sum -= cur;
            return;
        }
        for(int i = cur+1;i < 10;i++) {
            dfs(result,temp,i,sum,index+1,k,n);
        }
        temp.pop_back();
        sum -= cur;
        return;
    }
};

我的解答:

class Solution {
public:
    vector > combinationSum3(int k, int n) 
    {
        vector > res;
        vector path;
        int a=1,d=1;
        dfs(res,path,k,n,a,d);
        return res;
    }
    void dfs(vector >& res,vector& path,int k,int n,int a,int d)
    {
        if(n9)
        {
            return;
        }
        if(d==k&&n<10)
        {
            path.push_back(n);
            res.push_back(path);
            path.pop_back();
            return;
        }
        while(a<10)
        {
            path.push_back(a);
            dfs(res,path,k,n-a,a+1,d+1);
            path.pop_back();
            a++;
        }
    }
};

#发现一个错误测试用例:

1

10

你可能感兴趣的:(数据结构,leetcode,算法,LeetCode题解)