python3實現在二叉樹中找出和為某一值的所有路徑(推薦)

阿新 • • 發佈：2020-01-09

請寫一個程式建立一棵二叉樹，並按照一定規則，輸出二叉樹根節點到葉子節點的路徑。

規則如下：

1、從最頂端的根結點，到最下面的葉子節點，計算路徑通過的所有節點的和，如果與設定的某一值的相同，那麼輸出這條路徑上的所有節點。

2、從根節點遍歷樹時，請請按照左到右遍歷，即優先訪問左子樹的節點。

二叉樹建立規則：從上到下一層一層的，按照從左到右的順序進行構造

輸入"10,5,12,4,7"值，構造的樹如下：

1） 10
2） 10
/
5

3） 10
/\
5 12
4） 10
/\
5 12
/
4

5） 10
/\
5 12
/\
4 7

針對上面的二叉樹，如果當前我們設定的“路徑和”為19，那麼輸出結果為:

10,4

如果有多個路徑，按到左到右的順序遍歷生成的結果每行顯示一個顯示。例如如果當前我們設定的“路徑和”為22，那麼

輸出結果為:

10,7
10,12

如果沒有找到路徑和為設定的值的路徑，輸出error。

三、輸入:

輸入整數N---路徑和
一行字串，多個正整數，之間用","隔開

四、輸出: 滿足條件的二叉樹路徑

五、樣例輸入:

22
10,7

六、樣例輸出:

10,12

demo：

class Node(object):
  def __init__(self,x):
   self.val = x
   self.left = None
   self.right = None
class Tree(object):
 lt = [] # 依次存放左右孩子未滿的節點
 def __init__(self):
   self.root = None
 def add(self,number):
   node = Node(number) # 將輸入的數字節點化，使其具有左右孩子的屬性
   if self.root == None:
     self.root = node
     Tree.lt.append(self.root)
   else:
     while True:
       point = Tree.lt[0] # 依次對左右孩子未滿的節點分配孩子
       if point.left ==None:
         point.left = node
         Tree.lt.append(point.left) # 該節點後面作為父節點也是未滿的，也要加入到列表中。
         return
       elif point.right ==None:
         point.right = node
         Tree.lt.append(point.right) # 與左孩子同理
         Tree.lt.pop(0) # 表示該節點已擁有左右孩子，從未滿列表中去除
         return
class Solution:
 def __init__(self):
   self.results = []
 def RecursionFindPath(self,root,expectNumber,result):
   result.append(root.val)
   if root.left == None and root.right == None and sum(result) == expectNumber:
     self.results.append(result)
   temp = result[:]
   if root.left:
     self.RecursionFindPath(root.left,result)
   result = temp[:]
   if root.right:
     self.RecursionFindPath(root.right,result)
 def FindPath(self,expectNumber):
   if root == None:
     return []
   self.RecursionFindPath(root,[])
   self.results = sorted(self.results,key=len,reverse=True)
   return self.results
if __name__ =='__main__':
 t = Tree() # 二叉樹類的例項化
 L = [10,7]
 for i in L:
   t.add(i)
 expectNum = 22
 print(Solution().FindPath(t.root,expectNum))

輸出樣例：