1에서 9까지의 숫자만 사용할 수 있다는 점을 감안할 때 n이 되는 k 숫자의 가능한 모든 조합을 생성해야 한다고 생각하십시오. 각 조합은 고유한 숫자 집합이어야 합니다. 모든 숫자는 양수여야 하며 솔루션에는 중복 조합이 포함되어서는 안 됩니다. 따라서 k =3이고 n =9이면 가능한 조합은 [[1,2,6],[1,3,5],[2,3,4]]
입니다.이 문제를 해결하기 위해 다음 단계를 따릅니다. −
- solve라는 메서드를 형성하여 이 문제를 해결한다고 가정해 보겠습니다. 이것은 재귀적 방법이 될 것이며 k, n, 임시 배열 및 시작이 필요합니다. 시작은 처음에 1입니다.
- n =0인 경우
- temp의 크기가 k이면 res에 temp를 삽입하고 반환합니다.
- for i :=최소 9부터 시작하고 n
- temp에 i 삽입
- 해결(k, n – i, 온도, i + 1)
- temp에서 마지막 요소 삭제
- 주요 방법은 다음과 같습니다.
- temp라는 빈 벡터 하나 만들기
- 해결(k, n, 온도)
- 반환 결과
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
예시
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<vector<auto> > v){
cout << "[";
for(int i = 0; i<v.size(); i++){
cout << "[";
for(int j = 0; j <v[i].size(); j++){
cout << v[i][j] << ", ";
}
cout << "],";
}
cout << "]"<<endl;
}
class Solution {
public:
vector < vector <int> > res;
void solve(int k, int n, vector <int> temp, int start = 1){
if(n == 0){
if(temp.size() == k){
res.push_back(temp);
}
return;
}
for(int i = start ; i <= min(9, n); i++){
temp.push_back(i);
solve(k, n - i, temp, i + 1);
temp.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
res.clear();
vector <int> temp;
solve(k, n, temp);
return res;
}
};
main(){
Solution ob;
print_vector(ob.combinationSum3(2, 9));
} 입력
3 9
출력
[[1, 8, ],[2, 7, ],[3, 6, ],[4, 5, ],]