LeetCode 889. 根據(jù)前序和后序遍歷構(gòu)造二叉樹(shù)
返回與給定的前序和后序遍歷匹配的任何二叉樹(shù)惑淳。
pre 和 post 遍歷中的值是不同的正整數(shù)克伊。
示例：
輸入：pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1]
輸出：[1,2,3,4,5,6,7]
提示：
- 1 <= pre.length == post.length <= 30
- pre[] 和 post[] 都是 1, 2, ..., pre.length 的排列
- 每個(gè)輸入保證至少有一個(gè)答案片择。如果有多個(gè)答案，可以返回其中一個(gè)曙求。
遞歸
- 前序遍歷：根、左熬丧、右
- 后序遍歷：左糕簿、右、根
因此前序遍歷的左子樹(shù)根節(jié)點(diǎn)為 pre[1] 震檩。題目已知 pre 和 post 遍歷中的值是不同的正整數(shù)琢蛤。所以我們可以根據(jù) pre[1] 和值找到后序遍歷中左子樹(shù)根節(jié)點(diǎn)的位置。
這樣我們就能得到左抛虏、右子樹(shù)的前序和后序遍歷序列博其，并通過(guò)遞歸的方式利用左子樹(shù)的序列構(gòu)造出左子樹(shù)，利用右子樹(shù)的序列構(gòu)造出右子樹(shù)迂猴。
class Solution {
public TreeNode constructFromPrePost(int[] pre, int[] post) {
int len = pre.length;
if (len == 0) return null;
TreeNode root = new TreeNode(pre[0]);
if (len == 1) return root;
int index = 0; // 左子樹(shù)根節(jié)點(diǎn)下一個(gè)位置
for (int i = 0; i < len; i++) { // 找到后序遍歷中左子樹(shù)根節(jié)點(diǎn)的位置慕淡，先序遍歷中左子樹(shù)根節(jié)點(diǎn)位置為1
if (post[i] == pre[1]) {
index = i + 1;
break;
}
}
// 利用左子樹(shù)的前序和后序遍歷序列構(gòu)造出左子樹(shù)
root.left = constructFromPrePost(Arrays.copyOfRange(pre, 1, index + 1), Arrays.copyOfRange(post, 0, index));
// 利用右子樹(shù)的前序和后序遍歷序列構(gòu)造出右子樹(shù)
root.right = constructFromPrePost(Arrays.copyOfRange(pre, index + 1, len), Arrays.copyOfRange(post, index, len - 1));
return root;
}
}
- 時(shí)間復(fù)雜度：O(n2)
- 空間復(fù)雜度：O(n2)
