重建二叉樹
- 2019 年 10 月 4 日
- 筆記
本文來自sunandstarws的csdn文章
部落格地址:https://blog.csdn.net/sunandstarws/article/details/86578080
題目:
輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序遍歷序列{4,7,2,1,5,3,8,6},則重建二叉樹並返回。
思路:
看到這個題目就明白需要花點時間,上數據結構課的時候,明白了怎麼根據前序和中序可以唯一確定一顆二叉樹,並且也知道怎麼重建,但是要用程式碼實現起來,還是有點困難,畢竟自己很久沒有寫了。我知道要實現那種思路,肯定需要借用遞歸或者歸併。但是不是理的很清,後面覺得借鑒別人怎麼實現的,看完之後,決定自己動手,最開始用java編寫,寫完之後,出現空指針異常和數組非法訪問錯誤,與別人程式碼對了很久,最後終於找到錯誤了,這個題困了我一天。對於空指針異常,自己建立對象或者聲明數組的時候最好還是初始化,不要等到後面才想起,數組非法訪問是由於自己聲明的數組太大,沒有很嚴謹,導致我前面一直找不到自己錯在哪,感覺沒有越界,把數組大小嚴格改到題目要求所需之後,就提交成功了,可能之前寫C語言寫習慣了,沒有注意這方面,以後還是嚴謹一點。用js程式碼實現比Java要方便很多,比較順利,雖然也出現錯誤,但是解決了java坑之後,來解決Js的坑就快很多了;
對於這種題目,主要還是想清楚怎麼用遞歸實現,注意細節
java實現程式碼:
public class Solution { public TreeNode reConstructBinaryTree(int [] pre,int [] in) { int len=in.length; if(len==0) return null; TreeNode tree=new TreeNode(pre[0]); int i,temp=-1,j=0; for(i=0;i<len;i++){ if(pre[0]==in[i]){ temp=i; break; } } int[] left=new int[temp]; int[] leftpre=new int[temp]; for(i=0;i<temp;i++){ left[i]=in[i]; leftpre[i]=pre[i+1]; }//左子樹 int[] right=new int[len-temp-1]; int[] rightpre=new int[len-temp-1]; for(i=temp+1;i<len;i++){ right[j]=in[i]; rightpre[j]=pre[i]; j++; }//右子樹 tree.left=reConstructBinaryTree(leftpre,left); tree.right=reConstructBinaryTree(rightpre,right); return tree; }
JavaScript實現程式碼:
/* function TreeNode(x) { this.val = x; this.left = null; this.right = null; } */ function reConstructBinaryTree(pre, vin) { var len=vin.length; if(len==0) return null; var tree=new TreeNode(pre[0]); var i,temp=-1,j=0; for(i=0;i<len;i++){ if(pre[0]==vin[i]){ temp=i; break; } } var left=new Array(temp); var leftpre=new Array(temp); for(i=0;i<temp;i++){ left[i]=vin[i]; leftpre[i]=pre[i+1]; }//左子樹 var right=new Array(len-temp-1); var rightpre=new Array(len-temp-1); for(i=temp+1;i<len;i++){ right[j]=vin[i]; rightpre[j]=pre[i]; j++; }//右子樹 tree.left=reConstructBinaryTree(leftpre,left); tree.right=reConstructBinaryTree(rightpre,right); return tree; // write code here }