建立二叉树,并实现二叉树的遍历,先序遍历采用递归算法实现,层次遍历用非递归算法来实现。c++实现
时间: 2024-03-09 14:47:57 浏览: 64
好的,以下是C++实现的代码。
```cpp
#include <iostream>
#include <queue>
using namespace std;
// 定义节点结构体
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
// 先序遍历(递归实现)
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
if (!root) {
return res;
}
res.push_back(root->val);
vector<int> left = preorderTraversal(root->left);
res.insert(res.end(), left.begin(), left.end());
vector<int> right = preorderTraversal(root->right);
res.insert(res.end(), right.begin(), right.end());
return res;
}
// 层次遍历(非递归实现)
vector<int> levelOrderTraversal(TreeNode* root) {
vector<int> res;
if (!root) {
return res;
}
queue<TreeNode*> q;
q.push(root);
while (!q.empty()) {
TreeNode* node = q.front();
q.pop();
res.push_back(node->val);
if (node->left) {
q.push(node->left);
}
if (node->right) {
q.push(node->right);
}
}
return res;
}
// 测试代码
int main() {
// 创建二叉树
TreeNode* root = new TreeNode(1);
root->left = new TreeNode(2);
root->right = new TreeNode(3);
root->left->left = new TreeNode(4);
root->left->right = new TreeNode(5);
// 先序遍历
vector<int> preOrder = preorderTraversal(root);
for (int i = 0; i < preOrder.size(); i++) {
cout << preOrder[i] << " ";
}
cout << endl; // 输出:1 2 4 5 3
// 层次遍历
vector<int> levelOrder = levelOrderTraversal(root);
for (int i = 0; i < levelOrder.size(); i++) {
cout << levelOrder[i] << " ";
}
cout << endl; // 输出:1 2 3 4 5
return 0;
}
```
以上就是C++实现的代码,和Python实现的代码类似,只是语法不同。
阅读全文