zoukankan      html  css  js  c++  java
  • 剑指 Offer 07. 重建二叉树

    思路

    首先回忆下,用前序遍历和中序遍历一颗二叉树:

    	      1
    	    /   
    	   2     3
    	  /    /  
    	 4   5 6   7
    

    前序遍历的结果是:[1,2,4,5,3,6,7]
    中序遍历的结果是:[4,2,5,1,6,3,7]
    前序遍历的特点是,根节点始终出现在数组的第一位,而中序遍历中根节点出现在数组的中间位置。
    根据上面给出的两个数组,首先我们就可以拼出根节点,它就是1
    题目上已说明数组中不存在重复元素,那么由1就可以定位到中序数组的中间位置,中序数组中1左边的部分就是左子树,1右边部分就是右子树。

    前序数组怎么切分呢?注意看下面这张图,根节点是橘色,绿色部分是左子树,蓝色部分是右子树。

    前序数组的左子树部分+根节点1,2,4,5,中序数组的左子树部分+根节点4,2,5,1。这两者的数组长度是一样的。

    我们可以根据中序数组的中间位置1,来确定前序数组的左右部分,由于前序数组第一个是根节点,
    所以其左边部分是:[1:mid_index],右半部分是[mid_index+1:]
    这里的mid_index是中序数组的中间下标位置。
    递归函数实现如下:

    1. 终止条件:前序和中序数组为空
    2. 根据前序数组第一个元素,拼出根节点,再将前序数组和中序数组分成两半,递归的处理前序数组左边和中序数组左边,递归的处理前序数组右边和中序数组右边。

    动画演示如下:

     

    代码实现

     1 /**
     2  * Definition for a binary tree node.
     3  * struct TreeNode {
     4  *     int val;
     5  *     TreeNode *left;
     6  *     TreeNode *right;
     7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     8  * };
     9  */
    10 class Solution {
    11 public:
    12     TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
    13         if(preorder.empty())
    14             return NULL;
    15         
    16         return build(preorder, inorder, 0, preorder.size()-1, 0, inorder.size()-1);
    17     }
    18 
    19     TreeNode* build(vector<int>& preorder, vector<int>& inorder, 
    20         int preL, int preR, int inL, int inR) {
    21             if(preL > preR || inL > inR) {
    22                 return NULL;
    23             }
    24 
    25             int i;
    26             for(i = inL; i < inorder.size(); ++i) 
    27                 if(inorder[i] == preorder[preL]) 
    28                     break;
    29 
    30             TreeNode* root = new TreeNode(preorder[preL]);
    31             root->left = build(preorder, inorder, preL+1, preL+(i-inL), inL, i-1);
    32             root->right = build(preorder, inorder, preL+(i-inL)+1, preR, i+1, inR);
    33             return root;
    34 
    35     }
    36 };

    复杂度分析

  • 相关阅读:
    php连接mysql数据库基础
    控制操作
    巨慢IE9的加速
    推广邮件客户端(二):完美的IMAP客户端
    XPath 和 LINQ to XML 的比较
    推广邮件客户端(三):常用IMAP客户端介绍
    关于Git工具与GitHub
    Android开源项目(非组件)
    windows下使用Git获取Android源码
    Eclipse导入项目:No projects are found to import
  • 原文地址:https://www.cnblogs.com/FengZeng666/p/13846890.html
Copyright © 2011-2022 走看看