建立一棵二叉树并对其进行中序遍历

最新推荐文章于 2024-10-04 19:38:20 发布

原创最新推荐文章于 2024-10-04 19:38:20 发布 · 900 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#蓝桥杯 #c++ #算法

c++ 专栏收录该内容

23 篇文章

订阅专栏

这篇博客介绍了如何使用C++构建二叉树，并通过递归方法实现中序遍历。代码示例展示了如何从数组构建二叉树，以及遍历并输出中序遍历的结果。此外，还提供了主函数进行测试，输出了中序遍历的节点值。

#include <iostream>
using namespace std;
#include <vector>


struct TreeNode {
      int val;
      TreeNode *left;
      TreeNode *right;
      TreeNode() : val(0), left(nullptr), right(nullptr) {}
      TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
      TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
  };
 
TreeNode* newTreeNode(int data) {
    TreeNode* t = new TreeNode;
    t->val = data;
    t->left = nullptr;
    t->right = nullptr;
    return t;
}

TreeNode* creatTree(vector<int>& nums, int n, int index) {
    TreeNode* node = nullptr;
    if (index < n && nums[index] != -1) {
        node = newTreeNode(nums[index]);
        node->left = creatTree(nums, n, 2 * index + 1);
        node->right = creatTree(nums, n, 2 * index + 2);
    }
    return node;
}

//递归法对二叉树进行中序遍历
vector<int> ret;
void helper(TreeNode* root) {
    if (root == nullptr) return;
    helper(root->left);
    ret.push_back(root->val);
    helper(root->right);
}

vector<int> midOrder(TreeNode* root) {
    if (root == nullptr) return ret;
    helper(root);
    return ret;
}

int main() {
    vector<int> nums = { 2,3,4,10,-1 };
    TreeNode* root = creatTree(nums, nums.size(), 0);
    vector<int> nums1 = midOrder(root);
    for (int x : nums1) {
        cout << x << endl;
    }
    system("pause");
    return 0;
}