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