从前序与中序遍历序列构造二叉树(c++)

文章介绍了如何通过前序遍历和中序遍历的元素构建二叉树的递归函数,详细解释了如何找到根节点并递归构建左右子树的过程。
摘要由CSDN通过智能技术生成

题目描述如图,就是用提供的前序遍历(preorder)和中序遍历(inorder)来构造一个二叉树,由此可知:

前序遍历 :根  左子树  右子树

中序遍历 :左子树  根  右子树

我们可以构建一个函数 TreeNode* _buildTree(vector<int>& preorder, vector<int>& inorder,int& prei,int inbegin,int inend)  来进行递归调用这个二叉树的左子树和右子树,其中prei是判断每次根节点在preorder的下标元素

那么知道前序遍历时 第一个元素肯定是这个二叉树的根,就可以new一个根节点为TreeNode* root=new TreeNode(preorder[prei]);

 while(rooti<=inend)
        {
            if(inorder[rooti]==preorder[prei])//在中序中找到前序的根
            break;
            else
            ++rooti;
        }

当我们在中序中找到根节点后,便开始对根的左子树和右子树进行递归构建

//找到根后,开始建立左子树和右子树
        // [inbegin,rooti-1] rooti [rooti+1,inend]
        if(inbegin<=rooti-1)
        root->left=_buildTree(preorder,inorder,++prei,inbegin,rooti-1);
        else
        root->left=nullptr;

        if((rooti+1)<=inend)
        root->right=_buildTree(preorder,inorder,++prei,rooti+1,inend);
        else
        root->right=nullptr;

可以看出,在 // [inbegin,rooti-1] rooti [rooti+1,inend]中rooti为根=0,[inbegin,rooti-1]是中序左子树区间,[rooti+1,inend]是中序右子树区间

在条件分别为if(inbegin<=rooti-1)和if((rooti+1)<=inend)进行递归调用完成左,右子树的构建。

/**
 * Definition for a binary tree node.
 * 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) {}
 * };
 */
class Solution {
public:
    TreeNode* _buildTree(vector<int>& preorder, vector<int>& inorder,int& prei,int inbegin,int inend) {
        TreeNode* root=new TreeNode(preorder[prei]);

        int rooti=inbegin;
        while(rooti<=inend)
        {
            if(inorder[rooti]==preorder[prei])//在中序中找到前序的根
            break;
            else
            ++rooti;
        }

        //找到根后,开始建立左子树和右子树
        // [inbegin,rooti-1] rooti [rooti+1,inend]
        if(inbegin<=rooti-1)
        root->left=_buildTree(preorder,inorder,++prei,inbegin,rooti-1);
        else
        root->left=nullptr;

        if((rooti+1)<=inend)
        root->right=_buildTree(preorder,inorder,++prei,rooti+1,inend);
        else
        root->right=nullptr;

        return root;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int prei=0;
        int inbegin=0,inend=inorder.size()-1;

        return _buildTree(preorder,inorder,prei,inbegin,inend);
    }
};

  • 7
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
可以的,以下是实现二叉树的基本操作的 C 代码: #include <stdio.h> #include <stdlib.h> // 二叉树的定义 typedef struct BinaryTreeNode { int data; struct BinaryTreeNode *left; struct BinaryTreeNode *right; } BinaryTreeNode; // 根据扩展二叉树的前遍历序列建立二叉树 BinaryTreeNode* createBinaryTree1() { BinaryTreeNode *p; int ch; scanf("%c", &ch); if (ch == ' ') { p = NULL; } else { p = (BinaryTreeNode*)malloc(sizeof(BinaryTreeNode)); p->data = ch; p->left = createBinaryTree1(); p->right = createBinaryTree1(); } return p; } // 根据二叉树的前中序遍历序列构造二叉树 BinaryTreeNode* createBinaryTree2(int *preorder, int *inorder, int len) { if (len == 0) { return NULL; } BinaryTreeNode *p = (BinaryTreeNode*)malloc(sizeof(BinaryTreeNode)); p->data = preorder[0]; int i; for (i = 0; i < len; i++) { if (inorder[i] == preorder[0]) { break; } } int leftlen = i; int rightlen = len - i - 1; p->left = createBinaryTree2(preorder + 1, inorder, leftlen); p->right = createBinaryTree2(preorder + 1 + leftlen, inorder + leftlen + 1, rightlen); return p; } // 二叉树的前遍历 void preOrder(BinaryTreeNode *p) { if (p != NULL) { printf("%d ", p->data); preOrder(p->left); preOrder(p->right); } } // 二叉树中序遍历 void inOrder(BinaryTreeNode *p) { if (p != NULL) { inOrder(p->left); printf("%d ", p->data); inOrder(p->right); } } // 二叉树的后遍历 void postOrder(BinaryTreeNode *p) { if (p != NULL) { postOrder(p->left); postOrder(p->right); printf("%d ", p->data); } } int main() { // 手动输入前遍历序列建立二叉树 BinaryTreeNode *root1 = createBinaryTree1(); // 根据前中序遍历序列构造二叉树 int preorder[] = {1, 2, 4, 5, 3, 6, 7}; int inorder[] = {4, 2, 5, 1, 6, 3, 7}; BinaryTreeNode *root2 = createBinaryTree2(preorder, inorder, sizeof(preorder) / sizeof(int)); // 二叉树的遍历 printf("二叉树1的前遍历:"); preOrder(root1); printf("\n"); printf("二叉树1的中序遍历:"); inOrder(root1); printf("\n"); printf("二叉树1的后遍历:"); postOrder(root1); printf("\n"); printf("二叉树2的前遍历:"); preOrder(root2); printf("\n"); printf("二叉树2的中序遍历:"); inOrder(root2); printf("\n"); printf("二叉树2的后遍历:"); postOrder(root2); printf("\n"); return 0; }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值