刷题常用的二叉树代码模板

最新推荐文章于 2023-01-28 12:27:18 发布

HT_逍遥

最新推荐文章于 2023-01-28 12:27:18 发布

阅读量105

点赞数

分类专栏：数据结构和算法文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/qq_39256469/article/details/126113574

版权

数据结构和算法专栏收录该内容

4 篇文章 0 订阅

订阅专栏

二叉树结点定义

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) {}
};

二叉树的构造

层序序列构造

输入完全二叉树的序列，空结点位置用特殊字符标记，输出构造好的二叉树。
基于队列 + bfs即可。

前序序列和中序序列构造

	//函数定义：根据前序序列preorder[preStart, preEnd] 和 中序序列inorder[inStart, inEnd]来构造对应的二叉树并返回根结点
    TreeNode* build(vector<int>& preorder, int preStart, int preEnd, vector<int>& inorder, int inStart, int inEnd){
        if(preStart > preEnd || inStart > inEnd) return nullptr;//区间非法
        int rootVal = preorder[preStart];//根节点就是前序序列的首元素
        int index = 0;
        for(int i = inStart; i <= inEnd; i++){//获取根节点在中序序列中的位置
            if(inorder[i] == rootVal){
                index = i;
                break;
            }
        }
        int leftSize = index - inStart;
        //构造根节点
        TreeNode *root = new TreeNode(rootVal);
        //基于区间划分递归构造左子树和右子树
        root->left = build(preorder, preStart + 1,preStart + leftSize, inorder, inStart, index - 1);
        root->right = build(preorder, preStart + leftSize + 1, preEnd, inorder, index + 1, inEnd);
        return root;
    }

后序序列和中序序列构造

	TreeNode *build(vector<int>& inorder,int inStart,int inEnd, vector<int>& postorder,int postStart,int postEnd){
        if(inStart>inEnd || postStart>postEnd)
            return nullptr;
        int rootVal = postorder[postEnd];
        int index = 0;
        for(int i = inStart;i <= inEnd;i++){
            if(inorder[i] == rootVal)
                index = i;
        }
        int leftSize = index - inStart;
        TreeNode *root = new TreeNode(rootVal);
        root->left = build(inorder,inStart,index - 1,postorder,postStart,postStart + leftSize -1);
        root->right = build(inorder,index + 1,inEnd,postorder,postStart + leftSize,postEnd - 1);
        return root;
    }