劍指offer:22. 23. 24 (12.27)
阿新 • • 發佈:2019-01-05
''' 22.從上往下列印二叉樹 **題目:**從上往下打印出二叉樹的每個節點,同層節點從左至右列印。 **思路:**遞迴,每次將左子樹結果和右子樹結果存到結果集之中。 ''' # -*- coding:utf-8 -*- class TreeNode: def __init__(self, x): self.val = x self.left = None self.right = None class Solution: # 返回從上到下每個節點值列表,例:[1,2,3] def PrintFromTopToBottom(self, root): # write code here if root is None: return [] ans=[] ans.append(root.val) self.orderans(root,ans) return ans def orderans(self,root,ans): if not root: return if root.left: ans.append(root.left.val) if root.right: ans.append(root.right.val) self.orderans(root.left, ans) self.orderans(root.right,ans) if __name__=='__main__': solution=Solution() A1 = TreeNode(1) A2 = TreeNode(2) A3 = TreeNode(3) A4 = TreeNode(4) A5 = TreeNode(5) A1.left=A2 A1.right=A3 A2.left=A4 A2.right=A5 ans=solution.PrintFromTopToBottom(A1) print(ans)
''' 23.二叉樹的後續遍歷序列 **題目:**輸入一個整數陣列,判斷該陣列是不是某二叉搜尋樹的後序遍歷的結果。如果是則輸出Yes,否則輸出No。假設輸入的陣列的任意兩個數字都互不相同。 **思路:**二叉搜尋樹的特性是所有左子樹值都小於中節點,所有右子樹的值都大於中節點,遞迴遍歷左子樹和右子樹的值。 ''' # -*- coding:utf-8 -*- class Solution: def VerifySquenceOfBST(self, sequence): if not sequence: return False if len(sequence)==1: return True i=0 while sequence[i]<sequence[-1]: i=i+1 k=i #找到左子樹的點 for j in range(i,len(sequence)-1): if sequence[j]<sequence[-1]: return False leftsequence=sequence[:k] rightsequence=sequence[k:len(sequence)-1] #當K=i時已經不滿足了 if len(leftsequence)>0: a1=self.VerifySquenceOfBST(leftsequence) if len(rightsequence)>0: a2=self.VerifySquenceOfBST(rightsequence) return a1 and a2 if __name__=='__main__': solution=Solution() num=list(map(int,input().split(' '))) ans=solution.VerifySquenceOfBST(num) print(ans)
''' 24.二叉樹中和為某一值的路徑 **題目:**輸入一顆二叉樹的根節點和一個整數,打印出二叉樹中結點值的和為輸入整數的所有路徑。 路徑定義為從樹的根結點開始往下一直到葉結點所經過的結點形成一條路徑。(注意: 在返回值的list中,陣列長度大的陣列靠前)。 **思路:**利用遞迴的方法,計算加左子樹和右子樹之後的值,當引數較多是,可以將結果新增到函式變數之中。 ''' # -*- coding:utf-8 -*- class TreeNode: def __init__(self, x): self.val = x self.left = None self.right = None class Solution: # 返回二維列表,內部每個列表表示找到的路徑 def FindPath(self, root, expectNumber): # write code here if not root: return [] ans=[] path=[] self.dfs(root,expectNumber,ans,path) ans.sort() return ans def dfs(self,root,target,ans,path): if not root: return path.append(root.val) if root.left is None and root.right is None and target==root.val: ans.append(path[:]) if root.left: self.dfs(root.left,target-root.val,ans,path) if root.right: self.dfs(root.right,target-root.val,ans,path) path.pop() if __name__=='__main__': A1=TreeNode(10) A2=TreeNode(8) A3=TreeNode(12) A4=TreeNode(4) A5=TreeNode(2) A6=TreeNode(2) A1.left=A2 A1.right=A3 A2.left=A4 A2.right=A5 A5.left=A6 expectNumber=22 solution=Solution() ans=solution.FindPath(A1,expectNumber) print(ans)