Python ノードの全選択

Question

前提・実現したいこと

classに作った要素を全て選択する方法を知りたい
ここに質問の内容を詳しく書いてください。
二分探索を用いた時にたとえば2,5,3,4,1と順番に値を持つノードを二分木に入れてそれらが何番目に小さい要素なのかをカウントする関数を再起的に入れたいと思っております。このときに
for i in ●●
のようにclassのノードを全選択するコードの書き方を知りたいのですが何かしら方法がありましたらご教授ください
recountのような操作を組み込みたいのですが、どのようにすればいいのかわからなかったです。

class Bintree7(Bintree):
    def __init__(self,key=None,data=None,count=None):
        super().__init__(key=None,data=None)
        self.count=count
    def recount(self,count,m):
        if self.count>=m:
            self.count += 1
        if self.left != None:
            self.left.recount(count,m)
        if self.right != None:
            self.right.recount(count,m)
        return
    def insert(self,key,data):
        if self.key == None:
            self.key = key
            self.data = data
            self.count = 1
            return self
        if key < self.key:
            if self.left == None:
                m=self.count
                self.left = self.__class__(key,data)
                self.left.count=m
                return True
            else:
                return self.left.insert(key,data)
        elif key > self.key:
            if self.right == None:
                m=self.count

                self.right = self.__class__(key,data) # selfと同じクラスのオブジェクトを生成する
                self.right = m+1
                return True
            else:
                return self.right.insert(key,data)
        else: # key == self.key:
            return False

Comments

[退会済みユーザー]

大学か何かの課題か知りませんが、「classのノードを全選択するコードの書き方を知りたい」「recountのような操作を組み込みたい」ってことですけど、そもそも質問文記載のBintree7が、recount関数の組み込み以前に、きちんと動いていることは貴方の方で検証済みなんでしょうか？
その基本となる部分で動くコードを示してから「classのノードを全選択するコードの書き方を知りたい」っていう質問をしないと、二分探索木の基本的な部分の修正・作成まで丸投げしてるってことになるんじゃないですかね。

Answer 1

「classのノードを全選択する」と「recountのような操作」は、
同じではないのでよく検討して下さい。

想定してるリストは、「子ノードを全て」では有りませんか？

意図は違うかもしれませんが、「クラスのインスタンス全て」だった場合は、
複数の二分木を作った時に、共通の親を持たないので順序が不定になります。

---

for i in ●●のように

二分木をネストした長さ２のリストとみなすと解りやすくなるはず、
ネストした構造を平坦化するには、yield from が使えます。

親クラスで .left .right が正しく実装されていると仮定して

    def __iter__(self):

        yield self

        if self.left:
            yield from self.left

        if self.right:
            yield from self.right

Answer 2

class Bintree7(Bintree):
    instances = []
    def __init__(self,key=None,data=None,count=None):
        super().__init__(key=None,data=None)
        self.count=count
        Bintree7.instances.append(self)

としておいて

for i in Bintree7.instances:

でできます。

Comments

[teamikl]

「classのノードを全選択」を字面通りに解釈されたのだと思うけど、
「recountのような操作」にはなってませんよ。
クラス変数にインスタンスを溜めてるので、複数の根を持つ木を区別出来ないし、
取り出す順序が二分木の構造に関係なくノードの生成順になってるはずです。