LeetCode面試題07. 重建二叉樹

面試題07. 重建二叉樹

解題思路:

通過前序遍歷找根節點,找到根節點後,再中序遍歷中找當前根節點,找到後,左邊的節點是當前根節點的左子樹中的節點,右邊的節點是當前根節點右子樹的節點。遞歸建樹。

例如:

代碼:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
struct TreeNode* construct(int* preorder, int preleft,int preright, int* inorder, int inleft,int inright){
    int i,leftson;
    if(preleft>preright)return NULL;
    struct TreeNode* root=(struct TreeNode*)malloc(sizeof(struct TreeNode));
    root->val=preorder[preleft];
    for(i=inleft;i<=inright;i++){
        if(inorder[i]==root->val)break;
    }
    leftson=i-inleft;
    root->left=construct(preorder,preleft+1,preleft+leftson,inorder,inleft,i-1);
    root->right=construct(preorder,preleft+leftson+1,preright,inorder,i+1,inright);
    return root;
}

struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize){
    return construct(preorder, 0, preorderSize - 1, inorder, 0 ,inorderSize - 1);
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章