LeetCode[105] 从前序与中序遍历序列构造二叉树

给定两个整数数组 preorderinorder ,其中 preorder 是二叉树的先序遍历inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。

示例 1:

复制代码
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

示例 2:

复制代码
输入: preorder = [-1], inorder = [-1]
输出: [-1]

超级经典的题目,这道题相信很多人都会,也理解题意,给出的例子也能瞬间构建出来二叉树,但是真的写起来,我个人觉得不是那么容易。

首先要搞清楚二叉树的前序遍历,中序遍历,后续遍历的定义,然后找出规律,然后再开始解题。

  • 解题思路:

1.preorder[0]是root
2.在inorder中找到preorder[0]的索引idx,索引之前的为root->left,中序遍历为inorder[1,idx-1] 索引之后为root->right,中序遍历为inorder[idx+1,end]
3.根据2中的索引位置,确定root->left的节点数量leftnum和root->right的节点数量rightnum
4.在preorder中确定左右子树的前序遍历,root->left的前序遍历为preorder[1,leftnum],root->right的前序遍历为preorder[leftnum+1,end]
5.处理好边界和迭代终止条件

解法1:

cpp 复制代码
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size() == 0) return nullptr;
        TreeNode* root = new TreeNode(preorder[0]);
        return build(preorder, 0, preorder.size() - 1,
            inorder, 0, inorder.size() - 1);
    }
    TreeNode* build(vector<int>& preorder, int prestart, int preend,
                    vector<int>& inorder, int instart, int inend)
    {
        if(prestart > preend) return nullptr;
        int val = preorder[prestart];
        int idx = find(inorder.begin(), inorder.end(), val) - inorder.begin();
        int leftsize = idx - instart;
        TreeNode* root = new TreeNode(val);
        root->left = build(preorder, prestart+1, prestart+leftsize,
                           inorder, instart, idx-1);
        root->right  = build(preorder, prestart+leftsize+1, preend,
                             inorder, idx+1, inend);
        return root;
    }

解法2:从注释中应该可以看出来我调试了多少次*_*,思路没打开,主要是进行了vector的拷贝,导致内存占用很大,边界问题也没有搞的很清楚,导致有不少if判断,注释打开的话submit会超时超内存,关闭注释就可通过。

cpp 复制代码
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size() == 0) return nullptr;
        TreeNode* root = new TreeNode(preorder[0]);
        
        vector<int>::iterator iter = find(inorder.begin(), inorder.end(), preorder[0]);
        int leftnodenum = iter - inorder.begin();
        int rightnodenum = inorder.end() - iter - 1;
        //cout << leftnodenum << " " << rightnodenum << endl;
        
        vector<int> leftpreorder, rightpreorder;
        if(leftnodenum > 0)
            leftpreorder.assign(preorder.begin()+1, preorder.begin()+leftnodenum+1);
        if(rightnodenum > 0)
            rightpreorder.assign(preorder.begin()+leftnodenum+1, preorder.end());
        /*cout << "leftpreorder: " ;
        printvec(leftpreorder);
        cout << "rightpreorder: ";
        printvec(rightpreorder);*/

        /*vector<int> leftinorder, rightinorder;
        if(iter != inorder.begin())
            leftinorder.assign(inorder.begin(), iter+1);
        if(iter != inorder.end())
            rightinorder.assign(iter+1, inorder.end());
        /*cout << "leftinorder: ";
        printvec(leftinorder);
        cout << "rightinorder: ";
        printvec(rightinorder);*/
        //return nullptr;

        root->left = buildTree(leftpreorder, leftinorder);
        root->right = buildTree(rightpreorder, rightinorder);
        return root;
    }
相关推荐
墨尘游子21 分钟前
一文读懂循环神经网络—从零实现长短期记忆网络(LSTM)
人工智能·python·深度学习·神经网络·算法·机器学习·lstm
刚入坑的新人编程1 小时前
暑期算法训练.2
c++·算法
NullPointerExpection1 小时前
LLM大语言模型不适合统计算数,可以让大模型根据数据自己建表、插入数据、编写查询sql统计
数据库·人工智能·sql·算法·llm·llama·工作流
爱coding的橙子2 小时前
每日算法刷题Day49:7.16:leetcode 差分5道题,用时2h
算法·leetcode·职场和发展
写写闲篇儿2 小时前
经典算法之基数排序
算法
xindafu3 小时前
代码随想录算法训练营第五十天|图论part1
c语言·算法·图论
满分观察网友z3 小时前
从选择困难到最优策略:我如何用DP搞定“鱼和熊掌兼得”的排程难题(1751. 最多可以参加的会议数目 II)
后端·算法
满分观察网友z3 小时前
从一团乱麻到井然有序:我的海量任务调度“秘密武器”( 1353. 最多可以参加的会议数目)
算法
努力写代码的熊大3 小时前
随机链表的复制数据结构oj题(力口138)
数据结构·链表
m0_535064604 小时前
C++类模版与友元
java·c++·算法