题解 | #重建二叉树#
重建二叉树
http://www.nowcoder.com/practice/8a19cbe657394eeaac2f6ea9b0f6fcf6
import java.util.*;
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {
return rebuildTree(pre,0,pre.length-1,vin,0,vin.length-1);
}
TreeNode rebuildTree(int [] pre,int pl,int pr,int [] vin,int vl,int vr){
if(pl>pr || vl > vr){
return null;
}
TreeNode head = new TreeNode(pre[pl]);
int findNu = -1 ;
int length = 0;
for(int i = vl;i<=vr;i++){
if(vin[i] == pre[pl]){
length = i - vl ;
findNu = i;
break;
}
}
head.left = rebuildTree( pre, pl+1, pl+length, vin, findNu - length, findNu - 1);
head.right = rebuildTree(pre,pl+length+1,pr,vin,findNu+1,vr);
return head;
}
}
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {
return rebuildTree(pre,0,pre.length-1,vin,0,vin.length-1);
}
TreeNode rebuildTree(int [] pre,int pl,int pr,int [] vin,int vl,int vr){
if(pl>pr || vl > vr){
return null;
}
TreeNode head = new TreeNode(pre[pl]);
int findNu = -1 ;
int length = 0;
for(int i = vl;i<=vr;i++){
if(vin[i] == pre[pl]){
length = i - vl ;
findNu = i;
break;
}
}
head.left = rebuildTree( pre, pl+1, pl+length, vin, findNu - length, findNu - 1);
head.right = rebuildTree(pre,pl+length+1,pr,vin,findNu+1,vr);
return head;
}
}