輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。

① 題目:

輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序遍歷序列{4,7,2,1,5,3,8,6},則重建二叉樹並返回。

② 確定思路:

前序遍歷確定根,以及左右子樹的父節點。

中序遍歷確定左右子樹。

在確定根之後,進行左右子樹遞歸遍歷即可。

/**
 * 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 [] in) {
    if(pre.length == 0){return null;}//當長度爲0直接返回空樹
    TreeNode root=new TreeNode(pre[0]);//pre[0]是根值
	int len=pre.length;
	//len=1 說明是隻有根的樹
	if(len==1){
        root.left=null;
        root.right=null;
		return root;
	}
	//找到中序根位置index
	int rootnode=root.val;
	int i;
	for(i=0;i<len;i++){
		if(rootnode==in[i])
			break;
	}
        
    /**
        按照根的搜索方式,建立子樹的前序遍歷和中序遍歷,並進行遞歸調用
        */
        
	//創建左子樹
	if(i>0){
        
		int[] pre_left=new int[i];
		int[] in_left=new int[i];
		for(int j=0;j<i;j++){
			pre_left[j]=pre[j+1];
		}
		for(int j=0;j<i;j++){
			in_left[j]=in[j];
		}
		root.left=reConstructBinaryTree(pre_left,in_left);
	}else{
		root.left=null;
	}
	//創建右子樹
	if(len-i-1>0){
		int[] pre_right=new int[len-i-1];
		int[] in_right=new int[len-i-1];
		for(int j=i+1;j<len;j++){            //j=i+1,因爲i爲中序根節點位置,也爲前序遍歷左子樹的最後一個節點的位置
			pre_right[j-i-1]=pre[j];
			in_right[j-i-1]=in[j];
		}
		root.right=reConstructBinaryTree(pre_right,in_right);
	}else{
		root.right=null;
	}
	
	
    return root;
    }

}

 

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