พิจารณาว่าเราต้องสร้างการจัดหมู่ที่เป็นไปได้ทั้งหมดของตัวเลข k ตัวที่ผลรวมเท่ากับ n กำหนดว่าใช้ตัวเลข 1 ถึง 9 เท่านั้น แต่ละการจัดหมู่ควรเป็นชุดตัวเลขที่ไม่ซ้ำกัน ทุกตัวเลขควรเป็นบวก และคำตอบต้องไม่มีการจัดหมู่ซ้ำ ดังนั้นถ้า k = 3 และ n = 9 การจัดหมู่ที่เป็นไปได้คือ [[1,2,6],[1,3,5],[2,3,4]]
ในการแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -
- สมมติว่าเราจะแก้ปัญหานี้โดยใช้เมธอดที่เรียกว่า solve นี่จะเป็นเมธอด Recursive รับ k, n, temp array, และ start start เริ่มต้นที่ 1
- ถ้า n = 0
- ถ้าขนาดของ temp = k ให้แทรก temp ลงใน res และ return
- for i := start ถึง minimum ของ 9 และ n
- แทรก i ลงใน temp
- solve(k, n - i, temp, i + 1)
- ลบองค์ประกอบสุดท้ายออกจาก temp
- เมธอดหลักจะเป็นแบบนี้
- สร้างเวกเตอร์ว่างหนึ่งอันชื่อ temp
- solve(k, n, temp)
- return res
ลองดูการนำไปใช้ต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -
ตัวอย่างโค้ด
#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, ],]
