leetcode--Construct Binary Tree from Preorder and Inorder Traversal

来源:互联网 发布:日本对外贸易数据2015 编辑:程序博客网 时间:2024/06/06 03:18

Given preorder and inorder traversal of a tree, construct the binary tree.

Note:
You may assume that duplicates do not exist in the tree.


分类:二叉树

题意:由前序遍历,中序遍历生成二叉树


解法1:递归。将前序遍历和中序遍历分成两部分。对于前序遍历而言,数组第一个元素为根节点。在中序遍历中找到这个根节点,其左右两边则分别是左右子树的中序遍历。

计算左子树长度len(根据根节点index求出),然后对应于前序遍历,从根节点(数组第一个节点)开始,len长度之内,同样是左子树的前序遍历,剩下的是右子树的前序遍历。

这样我们又得到两个中序遍历和前序遍历,递归求解即可。

[java] view plain copy
  1. /** 
  2.  * Definition for a binary tree node. 
  3.  * public class TreeNode { 
  4.  *     int val; 
  5.  *     TreeNode left; 
  6.  *     TreeNode right; 
  7.  *     TreeNode(int x) { val = x; } 
  8.  * } 
  9.  */  
  10. public class Solution {  
  11.     public TreeNode buildTree(int[] preorder, int[] inorder) {  
  12.         return hepler(preorder, 0, preorder.length-1, inorder, 0, inorder.length-1);  
  13.     }  
  14.       
  15.     public TreeNode hepler(int[] preorder,int pstart,int pend,int[] inorder,int istart,int iend){  
  16.         if(pstart>pend) return null;  
  17.         if(pstart==pend) return new TreeNode(preorder[pstart]);  
  18.         int f = preorder[pstart];  
  19.         TreeNode t = new TreeNode(f);  
  20.         int mid = istart;  
  21.         for(;mid<=iend;mid++){  
  22.             if(f==inorder[mid]) break;  
  23.         }  
  24.         int len = mid-istart;  
  25.         t.left = hepler(preorder, pstart+1, pstart+len, inorder, istart, mid-1);  
  26.         t.right = hepler(preorder, pstart+len+1, pend, inorder, mid+1, iend);  
  27.         return t;  
  28.     }  
  29. }  

原文链接http://blog.csdn.net/crazy__chen/article/details/46484415

阅读全文
0 0
原创粉丝点击