给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。并生成c++代码
时间: 2024-01-23 12:47:30 浏览: 118
二叉树已知前序和中序遍历,求后序遍历的C++代码实现
4星 · 用户满意度95%
以下是基于递归的 c++ 代码实现:
```c++
#include <iostream>
#include <vector>
using namespace std;
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};
// 根据前序遍历和中序遍历构建二叉树
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder, int preStart, int inStart, int inEnd) {
if (preStart > preorder.size() - 1 || inStart > inEnd) {
return nullptr;
}
TreeNode* root = new TreeNode(preorder[preStart]);
int inIndex = 0;
for (int i = inStart; i <= inEnd; i++) {
if (inorder[i] == root->val) {
inIndex = i;
break;
}
}
root->left = buildTree(preorder, inorder, preStart + 1, inStart, inIndex - 1);
root->right = buildTree(preorder, inorder, preStart + inIndex - inStart + 1, inIndex + 1, inEnd);
return root;
}
// 计算二叉树高度
int getHeight(TreeNode* root) {
if (root == nullptr) {
return 0;
}
int leftHeight = getHeight(root->left);
int rightHeight = getHeight(root->right);
return max(leftHeight, rightHeight) + 1;
}
int main() {
vector<int> preorder{1, 2, 4, 5, 3, 6};
vector<int> inorder{4, 2, 5, 1, 3, 6};
TreeNode* root = buildTree(preorder, inorder, 0, 0, inorder.size() - 1);
cout << "The height of the binary tree is: " << getHeight(root) << endl;
return 0;
}
```
其中,`buildTree` 函数用于根据前序遍历和中序遍历构建二叉树,`getHeight` 函数用于计算二叉树高度。在 `main` 函数中,首先构建二叉树,然后调用 `getHeight` 函数计算二叉树的高度,并输出结果。
阅读全文