剑指Offer #04 重建二叉树(递归)
生活随笔
收集整理的這篇文章主要介紹了
剑指Offer #04 重建二叉树(递归)
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
題目來源:牛客網(wǎng)-劍指Offer專題
題目地址:重建二叉樹
題目描述
輸入某二叉樹的前序遍歷和中序遍歷的結(jié)果,請(qǐng)重建出該二叉樹。假設(shè)輸入的前序遍歷和中序遍歷的結(jié)果中都不含重復(fù)的數(shù)字。例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序遍歷序列{4,7,2,1,5,3,8,6},則重建二叉樹并返回。
題目解析
第 n+1n+1n+1 次遇到這種題目了,簡單寫一下思路吧。
- 根據(jù)前序遍歷的性質(zhì)確定根節(jié)點(diǎn),然后在中序遍歷中查詢根節(jié)點(diǎn)的位置。
- 這樣就得到了中序遍歷左右子樹的范圍了,接下來就是要求左子樹的節(jié)點(diǎn)的個(gè)數(shù)。
- 求得左子樹節(jié)點(diǎn)個(gè)數(shù)之后,我們就知道了前序遍歷中左右子樹的范圍了。
- 建樹的過程是遞歸的,我們要做的就是對(duì)左右子樹重復(fù)上述過程。
(上圖為一次求解過程的圖示)
想看更詳細(xì)解析的可以看這篇:L2-006 樹的遍歷-團(tuán)體程序設(shè)計(jì)天梯賽GPLT
/*** 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) {int r = pre.length - 1;TreeNode root = buildTree(pre, in, 0, r, 0, r);return root;}//[la,ra]為前序遍歷序列范圍,[lb,rb]為中序遍歷序列的范圍public TreeNode buildTree(int [] pre,int [] in,int la, int ra, int lb, int rb) {if (la > ra || lb > rb) {return null;}//num表示根節(jié)點(diǎn)的權(quán)值int num = pre[la];TreeNode rt = new TreeNode(num);//p表示根節(jié)點(diǎn)在中序遍歷序列中的位置int p = lb;while (p <= rb && in[p] != num) p++;//cnt表示左子樹節(jié)點(diǎn)的個(gè)數(shù)int cnt = p - lb;//遞歸建樹rt.left = buildTree(pre, in, la + 1, la + cnt, lb, p - 1);rt.right = buildTree(pre, in, la+ cnt + 1, ra, p + 1, rb);return rt;} }如果本文對(duì)你有所幫助,要記得點(diǎn)贊哦~
總結(jié)
以上是生活随笔為你收集整理的剑指Offer #04 重建二叉树(递归)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 剑指Offer #03 从尾到头打印链表
- 下一篇: 剑指Offer #05 用两个栈实现队列