C++ 中的组合求和 IV
假设我们不得不生成所有可能的 k 个数字组合,这些数字加起来等于 n,且只能使用数字 1 至 9。每个组合都应为一组唯一的数字。所有数字都应为正数,且解中不得包含重复的组合。那么,如果 k = 3,n = 9,可能的组合是 [[1,2,6],[1,3,5],[2,3,4]]
要解决这个问题,我们将按照以下步骤操作 -
- 假设我们将使用一个名为 solve 的方法来解决它。这将是一个递归方法,它将把 k、n、temp 阵列和 start 作为参数。start 最初为 1
- if n = 0
- if temp 的大小 = k,则将 temp 插入 res 并返回
- 对于 i := start 到 9 和 n 的最小值
- 将 i 插入 temp
- solve(k, n – i, temp, i + 1)
- 删除 temp 中的最后一个元素
- 主要方法如下
- 创建一个名为 temp 的空向量
- solve(k, n, temp)
- 返回 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, ],]
广告