輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。
例如輸入前序遍歷序列和中序遍歷序列,則重建二叉樹並返回。
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
' a test module '
'''解題思想,遞迴,令pre[0]為當前節點的值,在tin中左側的均為pre[0]左子樹的節點,右側為右子樹,pre中從1到temp為pre[0]左子樹,temp到len(pre)為右子樹,依次遞迴
'''__author__ = 'zhang shuai'
class
treenode:
def__init__
(self, x):
self.val = x
self.left = none
self.right = none
class
solution:
# 返回構造的treenode根節點
defreconstructbinarytree
(self, pre, tin):
# write code here
node = self.order(pre, tin)
return node
deforder
(self, pre, tin):
ifnot len(tin) or
not len(pre):
return
i = pre[0]
j = tin.index(i)
temp = len(pre) + 1
for index, value in enumerate(pre[1:], 1): # 如果沒有左子樹,temp取pre長度+1
if tin.index(value) > j:
temp = index
break
node = treenode(i)
node.left = self.order(pre[1:temp], tin[:j]) # pre
node.right = self.order(pre[temp:], tin[j + 1:])
return node
a = [1, 2, 4, 3, 5, 6]
b = [4, 2, 1, 5, 3, 6]
root = solution().reconstructbinarytree(a, b)
演算法改進:
牛客網上最優演算法:
演算法思想一樣,不過通過左右子樹節點個數來代替temp更加明確,也更簡潔。
# -*- coding:utf-8 -*-
# class treenode:
# def __init__(self, x):
# self.val = x
# self.left = none
# self.right = none
class
solution:
# 返回構造的treenode根節點
defreconstructbinarytree
(self, pre, tin):
# write code here
if len(pre) == 0:
return
none
if len(pre) == 1:
return treenode(pre[0])
else:
flag = treenode(pre[0])
flag.left = self.reconstructbinarytree(pre[1:tin.index(pre[0])+1],tin[:tin.index(pre[0])])
flag.right = self.reconstructbinarytree(pre[tin.index(pre[0])+1:],tin[tin.index(pre[0])+1:] )
return flag
二叉樹 重建二叉樹
問題 給定二叉樹的前序遍歷結果和中序遍歷結果,恢復出原二叉樹。假設二叉樹中的元素都不重複,給定二叉樹的前序遍歷序列,二叉樹的中序遍歷序列。看到此題,我首先想到的是尋找根節點,由前序遍歷序列可以看出根節點為1,此時通過中序遍歷可以看出來4,7,2在根節點的左子樹,5,3,8,6在樹的右節點。此時我們可...
二叉樹 重建二叉樹
題目給定兩個陣列,乙個是前序遍歷陣列 preorder 乙個是中序遍歷陣列 inorder 要求輸出還原二叉樹 核心在於我們要理解前序和中序便利的特點 前序遍歷 根節點 左節點 右節點 中序遍歷 左節點 根節點 右節點 所以我們從二叉樹的根節點開始重構 也就是preorder的第乙個值 同時用乙個m...
二叉樹重建
摘自劉汝佳的 演算法競賽入門經典 preorder t t 的根結點 preorder t 的左子樹 preorder t 的右子樹 inorder t inorder t 的左子樹 t 的根結點 inorder t 的右子樹 postorder t postorder t 的左子樹 postord...