重建二叉樹

2021-08-10 12:46:17 字數 2054 閱讀 6575

輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建出該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。

例如輸入前序遍歷序列和中序遍歷序列,則重建二叉樹並返回。

#!/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...