题解 | #牛棚分组#
牛棚分组
https://www.nowcoder.com/practice/d5740e4dde3740828491236c53737af4
考察的知识点:回溯;
解答方法分析:
- 创建一个空的二维 vector 结果 res,用于存储所有组合方案。
- 创建一个辅助函数 backtrack(int n, int k, int start, vector<int>&, vector<vector<int>>& res),用于回溯。
- 在 backtrack() 函数中,首先检查当前组合 path 的长度是否等于 k,即是否已经得到了一个满足牛的数量的组合。如果是,则将当前组合加入到结果 res 中,并返回。
- 如果 path 的长度还不够 k,那么从 start 位置开始,遍历数字的范围从 start 到 n。对于个数字 i,将其加入到当前组合 path 中,然后递归调用 backtrack() 函数来寻找下一个数字。
- 在递归调用完成后,需要将刚刚加入的数字 i 从 path 中移出,以便下一轮的尝试其他可能的选择。
- 最后,在 combine() 函数中,通过调用 backtrack() 函数来开始回溯。
- 返回最终的结果 res。
所用编程语言:C++;
完整编程代码:↓
class Solution { public: vector<vector<int>> combine(int n, int k) { vector<vector<int>> res; vector<int> path; backtrack(n, k, 1, path, res); return res; } void backtrack(int n, int k, int start, vector<int>& path, vector<vector<int>>& res) { if (path.size() == k) { res.push_back(path); return; } for (int i = start; i <= n; i++) { path.push_back(i); backtrack(n, k, i + 1, path, res); path.pop_back(); } } };