python-与えられたノードの親を印刷する方法

Question

私はバイナリツリーを作成しました。私は手動で渡しているノードの親を見つけようとしています。それは左のサブツリーではうまく動作しますが、右のサブツリーには特定のノードだけがあります。

# To create a tree from scratch 
class tree: 
    """To create nodes each time an instance has been 
    created""" 
    def __init__(self, key): 
     self.data = key 
     self.left = None 
     self.right = None 

    def parent_search(self, root, child_node): 
     if root : 
     if root.left.data== child_node: 
      return root.data 
     if root.right.data== child_node: 
      return root.data 
     elif root: 
      return self.parent_search(root.left, child_node) 
      return self.parent_search(root.right, child_node) 

root = tree(10) 
root.left = tree(20) 
root.left.left = tree(90) 
root.left.right = tree(100) 
root.left.left.left = tree(80) 
root.right = tree(30) 
root.right.left = tree(40) 
root.right.right = tree(50) 
print(root.parent_search(root,80)) 

私はroot.parent_search(root,80)を与える場合、私はそれは私にNonetypeエラーになります右に40を仮定検索する場合、私は80の親として90を得る.but。

return self.parent_search(root.left, child_node) 
    File "/home/vaibhav/Desktop/Data_Structures/python_play_area.py", line 14, in parent_search 
    if root.right.data== child_node: 
AttributeError: 'NoneType' object has no attribute 'data' 

観察 私は、我々はそれが親または子の電話の際root.left.rightが正常に渡されたばかり見ることができるので、サブツリーの右の子を持つ任意の問題を参照してくださいいけません。

Answer 1

なぜ2つのreturnを使用したのか分かりますが、関数から何かを返す必要があり、左右の配列の両方でこれを呼び出す必要があります。そのためにはorを使用してください。 、something or Noneは常にsomething

def parent_search(self, root, child_node): 
    if not root: return None 
    if root.left and root.left.data==child_node: return root 
    if root.right and root.right.data==child_node: return root 
    return self.parent_search(root.left, child_node) or self.parent_search(root.right, child_node) 

Answer 2

を返すので、私はちょうどroot.left ==子とそれが働いていた場合の条件でroot.left場合は追加することによって、自分のコードに小さな変更を行いました。これは私がやった最小限の変更です。

# To create a tree from scratch 
class tree: 
    """To create nodes each time an instance has been 
    created""" 
    def __init__(self, key): 
     self.data = key 
     self.left = None 
     self.right = None 

    def parent_search(self, root, child_node): 
     if root : 
      if root.left and root.left.data== child_node: 
       return root.data 
      if root.right and root.right.data== child_node: 
       return root.data 
      elif root: 
       return self.parent_search(root.left, child_node) or self.parent_search(root.right, child_node) 


root = tree(10) 
root.left = tree(20) 
root.left.left = tree(90) 
root.left.right = tree(100) 
root.left.left.left = tree(80) 
root.right = tree(30) 
root.right.left = tree(40) 
root.right.right = tree(50) 
print(root.parent_search(root,50))