题解 | #牛棚分组#
牛棚分组
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();
}
}
};
