建立二叉树,并实现二叉树的遍历,先序遍历采用递归算法实现,层次遍历用非递归算法来实现。c++实现,用户可输入
时间: 2024-03-09 13:48:03 浏览: 70
数据结构二叉树实验(C++).docx
5星 · 资源好评率100%
好的,以下是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) {}
};
// 建立二叉树
TreeNode* createTree() {
int val;
cin >> val;
if (val == -1) { // 输入-1表示该节点为空
return NULL;
}
TreeNode* root = new TreeNode(val);
root->left = createTree();
root->right = createTree();
return root;
}
// 先序遍历(递归实现)
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() {
// 创建二叉树
cout << "请输入二叉树的节点值,-1表示该节点为空:" << endl;
TreeNode* root = createTree();
// 先序遍历
vector<int> preOrder = preorderTraversal(root);
cout << "先序遍历结果为:";
for (int i = 0; i < preOrder.size(); i++) {
cout << preOrder[i] << " ";
}
cout << endl;
// 层次遍历
vector<int> levelOrder = levelOrderTraversal(root);
cout << "层次遍历结果为:";
for (int i = 0; i < levelOrder.size(); i++) {
cout << levelOrder[i] << " ";
}
cout << endl;
return 0;
}
```
以上就是C++实现的代码,可以让用户输入二叉树节点的值来建立二叉树,并实现了先序遍历和层次遍历。
阅读全文