LeetCode 216.組合總和 Ⅲ(C++)
阿新 • • 發佈:2018-11-19
找出所有相加之和為 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]]
執行用時為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