バイナリツリーを実装する際の問題

Question

class Node: 
    def __init__(self, v): 
     self.l = None 
     self.r = None 
     self.v = v 

class BinaryTree: 
    def __init__(self): 
     self.root = None 

    def put(self, v): 
     if self.root is None: 
      self.root = Node(v) 
     else: 
      if self.root is None: 
       self.root = Node(v) 
      elif self.root.v <= v: 
       self.root.r = self.put(self.root.r, v) 
      elif self.root.v > v: 
       self.root.l = self.put(self.root.l, v) 
     return self.root 

    def __contains__(self, v): 
     return finns(self.root, v) 

    def write(self): 
     if self.root is not None: 
      print(self.root.v) 
      if self.root.l is not None: 
       self.write(self.root.l) 
      if self.root.r is not None: 
       self.write(self.root.r) 

a = BinaryTree() 
a.put(3) 
a.put(4) 
a.write() 

なぜ動作しないのでしょうか？それは言う：

TypeError: put() takes 2 positional arguments but 3 were given 

私はちょうどツリーに入力整数にput()を使用します。あなたは2つの明示的な引数を持つインスタンスメソッドを呼び出すputself.root.r = self.put(self.root.r, v)ラインで

Answer 1

put()を間違って使用しているため、TypeErrorが届いています。しかし、本当の問題は、ツリーを適切に構築していないことです。

以下のコードでは、この問題は修正されていますが、私はwrite()__contains__()の両方の方法も修正しました。

class Node: 
    def __init__(self, v): 
     self.l = None 
     self.r = None 
     self.v = v 

class BinaryTree: 
    def __init__(self): 
     self.root = None 

    def put(self, v): 
     if self.root: 
      self._put(v, self.root) 
     else: 
      self.root = Node(v) 

    def _put(self, v, node): 
     if v < node.v: 
      if node.l: 
       self._put(v, node.l) 
      else: 
       node.l = Node(v) 
     else: 
      if node.r: 
       self._put(v, node.r) 
      else: 
       node.r = Node(v) 

    def __contains__(self, v): 
     return (False if not self.root 
        else self._find(v, self.root) is not None) 

    def _find(self, v, node): 
     if v == node.v: 
      return node 
     elif v < node.v and node.l is not None: 
      return self._find(v, node.l) 
     elif v > node.v and node.r is not None: 
      return self._find(v, node.r) 

    def write(self): 
     if self.root is not None: 
      self._write(self.root, 0) 

    def _write(self, node, level): 
     if node is not None: 
      self._write(node.l, level+1) 
      print(' '*level + str(node.v)) 
      self._write(node.r, level+1) 

a = BinaryTree() 
a.put(3) 
a.put(4) 
a.put(5) 
a.put(9) 
a.put(7) 
a.put(10) 
a.write() 
print('') 
print('{:2} in "a" -> {}'.format(5, 5 in a)) 
print('{:2} in "a" -> {}'.format(42, 42 in a)) 

出力：

3 
    4 
    5 
     7 
     9 
     10 

5 in "a" -> True 
42 in "a" -> False 

Answer 2

：

（「V」は値の略で、「R」、左右のための「L」のために注意してください。）。 selfでメソッドを呼び出すので、メソッドがバインドされ、selfが暗黙的に最初の引数として渡されます（合計3つの引数）。

現在のところ、あなたのコードでは、簡単に修正することはできません。たとえば、インスタンスをBinaryTree.putに明示的に渡すことはできますが、現在はNodeではなくBinaryTreeのインスタンスを渡そうとしています。