题解 | #重建二叉树#

重建二叉树

https://www.nowcoder.com/practice/8a19cbe657394eeaac2f6ea9b0f6fcf6

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <vector>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param preOrder int整型vector 
     * @param vinOrder int整型vector 
     * @return TreeNode类
     */
    TreeNode* rebuild(TreeNode* root,vector<int> preOrder,vector<int> vinOrder,int preL,int preR,int inL,int inR){
        if(preL>preR)
            return nullptr;

        root=new TreeNode(preOrder[preL]);
        int index;
        for(int i=inL;i<=inR;i++){
            if(vinOrder[i]==preOrder[preL]){
                index=i;
                break;
            }
        }
        int leftNum=index-inL;
        root->left=rebuild(root->left,preOrder,vinOrder,preL+1,preL+leftNum,inL,index-1);
        root->right=rebuild(root->right,preOrder,vinOrder,preL+leftNum+1,preR,index+1,inR);
        return root;
    }
    TreeNode* reConstructBinaryTree(vector<int>& preOrder, vector<int>& vinOrder) {
        // write code here
        TreeNode* root=rebuild(root,preOrder,vinOrder,0,preOrder.size()-1,0,vinOrder.size()-1);
        return root;

    }
};

根据推导结果,可以得到上面的递归表达

全部评论

相关推荐

刘湘_passion:出国旅游?那就小心你的腰子咯
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务