生成打印二叉树从根结点到叶子结点的所有路径的C++代码
时间: 2023-11-25 14:04:28 浏览: 82
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径
以下是生成打印二叉树从根节点到叶子节点的所有路径的C++代码:
```cpp
#include <iostream>
#include <vector>
using namespace std;
// 二叉树节点结构体定义
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
void printPath(vector<int>& path) {
for (int i = 0; i < path.size() - 1; i++) {
cout << path[i] << "->";
}
cout << path[path.size() - 1] << endl;
}
void binaryTreePathsHelper(TreeNode* root, vector<int>& path, vector<vector<int>>& res) {
if (!root) return;
path.push_back(root->val);
if (!root->left && !root->right) {
res.push_back(path);
printPath(path);
}
binaryTreePathsHelper(root->left, path, res);
binaryTreePathsHelper(root->right, path, res);
path.pop_back();
}
vector<vector<int>> binaryTreePaths(TreeNode* root) {
vector<vector<int>> res;
vector<int> path;
binaryTreePathsHelper(root, path, res);
return res;
}
int main() {
// 二叉树构建
TreeNode* root = new TreeNode(1);
root->left = new TreeNode(2);
root->right = new TreeNode(3);
root->left->right = new TreeNode(5);
// 打印二叉树从根节点到叶子节点的所有路径
binaryTreePaths(root);
return 0;
}
```
该代码通过递归遍历二叉树,将从根节点到当前节点的路径记录在一个 vector 中,并在遍历到叶子节点时将当前路径保存到结果集中。同时,我们通过 printPath 函数打印路径。最终,该程序会输出如下结果:
```
1->2->5
1->3
```
阅读全文