1. 程式人生 > >LeetCode 216.組合總和 Ⅲ(C++)

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<vector<int>> combinationSum3(int k, int n) {
        vector<vector<int>> result;
        if(n < 3 || n > 45) return result;
        vector<int> temp;
        for(int i = 1;i <= 9;i++) {
            dfs(result,temp,i,0,1,k,n);
        }
        return result;
    }
    void dfs(vector<vector<int>> & result,vector<int> &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<vector<int> > combinationSum3(int k, int n) 
    {
        vector<vector<int> > res;
        vector<int> path;
        int a=1,d=1;
        dfs(res,path,k,n,a,d);
        return res;
    }
    void dfs(vector<vector<int> >& res,vector<int>& path,int k,int n,int a,int d)
    {
        if(n<a||a>9)
        {
            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