根据前序遍历和中序遍历树构造二叉树

给出中序遍历:[1,2,3]和前序遍历:[2,1,3]. 返回如下的树:

2
/ \
1 3

补充知识:

前序排列:根节点,左子节点,右子节点
中序排列:左子节点,根节点,右子节点
这里写图片描述

解题思路:

1.根据前序排列第一个数字确定根节点,并在中序排列中确定根节点位置
2.根据根节点的位置确定中序排列中左子树、右子树,中序排列中根节点之前的数字为左子树,根节点之后的部分为右子树
3.根据中序排列中左子树和右子树的长度确定在前序排列中的左子树、右子树
4.递归构建左子树、右子树,直到叶子节点(判断条件)

TreeNode * buildTree(vector<int> &preorder, vector<int> &inorder) {
        // write your code here
        TreeNode *root = NULL;
        vector<int> preorder_l,preorder_r,inorder_l,inorder_r;
        int i,root_index=0;

        if(preorder.empty()!=1 || inorder.empty()!=1) {
            root = new TreeNode(preorder[0]); //  在前序队列中找根节点

            //  在中序队列中找出根节点位置
            for(i=0; i<inorder.size(); i++) {
                if(preorder[0] == inorder[i])
                    break;
                root_index++;
            }

            //  左右子树的前序、中序队列
            for(i=0; i<root_index; i++) {
                preorder_l.push_back(preorder[i+1]);
                inorder_l.push_back(inorder[i]);
            }
            for(i=root_index+1; i<inorder.size(); i++) {
                preorder_r.push_back(preorder[i]);
                inorder_r.push_back(inorder[i]);
            }
            //递归构建左右子树
            root->left = buildTree(preorder_l, inorder_l);
            root->right = buildTree(preorder_r, inorder_r);
        }
        return root;



    }

类型 *p=new 类型[];//分配新的内存,指针指向数组第一个元素

猜你喜欢

转载自blog.csdn.net/htt789/article/details/80164049