Bstをclassで作ろう

[b]このワークシートは[url=https://www.geogebra.org/m/twxxx3yq]Math by Code[/url]の一部です。[br][/b][br]
2分探索BS
2分探索木BST
1.2分探索
<2分探索>[br]データがソートされているとき、[br][br][color=#0000ff]探し物があるときに2分探索はとても便利だということは[br]だれでも経験があるでしょう。[br][/color][br]上か下かで調べる対象が[br]半分になっていくというやつですね。[br][br]N個数を半分にしていくときに最悪で何回半分にするのかが調べる回数になるね。[br]たとえば、[color=#0000ff]3回調べて見つかるとしたら,N=2[sup]3[/sup][br]となる。言い換えると3=log[sub]2[/sub]Nだね。[br][/color][br]だから、[color=#0000ff][b][size=150]調べる回数はlogN[/size][/b][/color]のレベルだということだ。[br]もしもNが1024だとしたら、2^10=1024だから、log1024=10だ。[br]前から順にさがして最悪1024回調べるところが、2分探索ではわずか最悪10回ですむ。[br][br]調べる回数が100分の1のレベルに激減するね。[br]すばらしい!
2.2分探索木(BST)
[color=#0000ff][b]2分探索BinarySearch, BS[/b][/color]はデータはソートされているだけだったが、[br]ソートするという手間がかかっているからこそ、検索が楽にできた。[br][br]では、2分探索という発想がそのままデータ構造になるとすごく便利になりそうだね。[br]それが、[b][color=#0000ff]2分探索木BinarySearchTree(BST)[/color][/b]だ。[br][br][color=#0000ff][b]BSTは親から左の部分木が親より小さく、親から右の部分木が親より大きく、これがどの点を親にしても[br]成り立つ。[br][/b][/color]だから、BSTそのものがソートをされたデータを表していることになる。[br]データxを追加するときは親と比べて大きかったら右の部分木に進み、親より小さかったら左に進む。[br]これを続けていけば、いつかは行き先が決まる。[br]データ検索も、データ削除も同じ要領でできるはずだね。[br][br][color=#9900ff][u][b][size=150]質問:pythonでBSTを実装するにはどうしたらよいでしょうか。[br][/size][/b][/u][/color][br]BSTの最小単位は左木と右木とデータをもつ頂点ですね。[br]その[b]3つのデータ(left, right,data)[/b]をもつ頂点を定義しよう。[br]たいていはnodeという名前で定義されている。[br]しかし、ただの点・節ではなく[u][b]2分木の最小単位なので「BT」[/b][/u]とした。[br][br]BSTはBTがrootから順々に下に伸びているものだね。[br][br][b]1件追加する関数insert(data)[/b]は行き先を探して見つかったら[b]BTを値としてセット[/b]するものだ。[br]根rootが空なら、[b]root=BT(data)[/b]とセットする。[br]空でないなら再帰的に今いる[b]v.data[/b]より入れたい[b]new_v.data[/b]が小さいと[b]v.left[/b]へ,[br]大きいなら[b]v.left[/b]へ登録しよう。[br]あとで、追加状態の確認ができるように、[br]左に追加したら、[color=#0000ff]print(f"{new_v.data} <={v.data}")[/color][br]右に追加したら、[color=#0000ff]print(f"{v.data}=> {new_v.data}")[/color][br]とすることで、木の登録状況の確認ができるね。[br][br]1件ずつ追加するのは大変だから、[br]リスト渡しでinsert(data)をくり返してくれる[b]関数[br]make_from(lst)[/b]を作ろう。[br][br]1件追加と同様のロジックで関数[b]seach(data)[/b]も作ろう。[br][br]#[IN]Python[br][color=#0000ff]#2分する頂点クラスBT==========================[br]# dataと左枝と右枝を持つデータ構造(dataは大小判断ができれば何でもよい。)[br][b]class BT:[/b][br] def __init__(self, data):[br] self.[b]left[/b] = None[br] self.[b]right[/b] = None[br] self.[b]data[/b] = data[br]#2分探索木のクラスBST==========================[br]# rootから順にBT(data)を入れる。[br]# rootを含む各点vで、v、v.left、v.rightにBT(data)のデータを入れ続けられます。[br][b]class BST:[br][/b] def __init__(self):[br] [b] self.root = None[/b][br][br] # リストから作る。[br] [b] def make_from(self, lst): [br][/b] for item in lst:[br] self.insert(item)[br] # 1件追加[br] [b]def insert(self, data):[br][/b] # vにBTデータをvに入れる。[br] # vといえば、BTデータとして読んでください。[br] v = BT(data) [br] if self.root is None:[br] self.root = v[br] else:[br] self._insert_recursive(self.root, v)[br] def _insert_recursive(self, v, new_v):[br] if new_v.data < v.data:[br] if v.left is None:[br][b] v.left = new_v[br] print(f"{new_v.data} <={v.data}")[br][/b] else:[br] self._insert_recursive(v.left, new_v)[br] elif new_v.data > v.data:[br] if v.right is None:[br][b] v.right = new_v[br] print(f"{v.data}=> {new_v.data}")[br][/b] else:[br] self._insert_recursive(v.right, new_v)[br] else:[br] print(f"データ{v.data}が重複しています。")[br] # 1件探す [br] [b]def search(self, data):[br][/b] return self._search_recursive(self.root, data)[br] def _search_recursive(self, v, data):[br] if v is None:[br] return False[br] if data == v.data:[br] return True[br] elif data < v.data:[br] return self._search_recursive(v.left, data)[br] else:[br] return self._search_recursive(v.right, data)[br][b][br]myBST=BST()[br]nums = [20,30,40,5,10,15,25,35][br]myBST.make_from(nums)[br][/b][/color]#===============================[br][OUT][br]20=> 30[br]30=> 40[br]5 <=20[br]5=> 10[br]10=> 15[br]25 <=30[br]35 <=40
3.BSTの視覚化
BSTを文字で表示しても、正しく登録されたかどうかだけしかわかりません。[br][br]グラフを使って、全体としての大小関係や全体として左右のバランスが見えるようにしたい。[br][br][color=#9900ff][b][u][size=150]質問:どうすれば、BSTのデータからグラフができるでしょう。[br][/size][/u][/b][/color][br]BSTにBTつないで、printしている部分を変えることを考えよう。[br]グラフオブジェクトを宣言して、[br][b]print(f"親=>子")、print(f"子<=親")の部分を[br][color=#0000ff][size=150]G.edge(親,子,label=左/右)[br][/size][/color]のようにかきかえて[br][/b]そのつど矢印と左か右かのラベルを登録してもよいかもしれない。[br]しかし、グラフは表示されるが問題がある。それは、[color=#ff00ff]登録した順に矢印が書かれるからだ[/color]。[br]左L,右rが正しくかかれていても、登録順が右が先で、左が後だと、左側にLの辺が、右側にrの辺が表示[br]されてしまう。[br]これでは、数が左から小さい順にならぶようには見えない。[br][br]そこで、深さdepthを辺に持たせた。[br]insert関数が再帰関数を使うときに深さを増やしていき、辺に深さを登録しておき、すぐには描画しない。[br]BSTクラスにedges情報をもたせて、辺情報を追加していこう。[br][br][b][color=#0000ff][size=150]edges.append( 深さ、始点、終点、Lかr)[br][/size][/color][/b][br]のように。[br]ついでに、[b]親と同じ数[/b]を登録しようとすると、はじかれるようにした。[br][br]そして、グラフを描く関数の中で、edgesの1つ1つをすぐに描画しない。[br][u][b][color=#0000ff]まず、edgesをソートする。前の数値からソートするので、[br][/color][/b][/u][b][u][color=#0000ff]深さ、始点、終点が小さい順にソートされる。[/color][/u][br]すると深さの小さい辺、始点が小さい辺、終点が小さい辺[/b][br]という順になる。[br]この順にしてから、graphvizに登録すると、[b]同じ深さでは小さい数が左にくる[/b]。[br][br]以下が改良後のコードだ。[br][b]#[IN]Python[br]#======================[br]import graphviz[br][br][/b]#2分する頂点クラスBT==========================[br]# dataと左枝と右枝を持つデータ構造(dataは大小判断ができれば何でもよい。)[br]class BT:[br] def __init__(self, data):[br] self.left = None[br] self.right = None[br] self.data = data[br]#2分探索木のクラスBST==========================[br]# rootから順にBT(data)を入れる。[br]# rootを含む各点vで、v、v.left、v.rightにBT(data)のデータを入れ続けられます。[br]class BST:[br] def __init__(self):[br] self.root = None[br] [b] self.edges =[][br][/b] # リストから作る。[br] def make_from(self, lst): [br] for item in lst:[br] self.insert(item)[br] # 1件追加[br] def insert(self, data):[br] # vにBTデータをvに入れる。[br] # vといえば、BTデータとして読んでください。[br] v = BT(data) [br] if self.root is None:[br] self.root = v[br] else:[br] self._insert_recursive(self.root, v, [b]0[/b])[br] def _insert_recursive(self, v, new_v, [b]depth[/b] ):[b][br] [color=#0000ff]depth = depth[br] depth +=1[br][/color] [/b] if new_v.data < v.data:[br] if v.left is None:[br] [b] v.left = new_v[br] [color=#0000ff] self.edges.append([depth,v.data, new_v.data, 'L' ])[br][/color][/b] else:[br] self._insert_recursive(v.left, new_v, [b]depth[/b])[br] elif new_v.data > v.data:[br] if v.right is None:[br] [b] v.right = new_v[color=#0000ff][br] self.edges.append([depth,v.data, new_v.data, 'r' ])[br][/color][/b] else:[br] self._insert_recursive(v.right, new_v, [b]depth[/b])[br] [b]else:[br] print(f"データ{v.data}が重複しています。")[br] [/b][br] # 1件探す [br] def search(self, data):[br] return self._search_recursive(self.root, data)[br] def _search_recursive(self, v, data):[br] if v is None:[br] return False[br] if data == v.data:[br] return True[br] elif data < v.data:[br] return self._search_recursive(v.left, data)[br] else:[br] return self._search_recursive(v.right, data)[br] [color=#0000ff][b] def show_graph(self):[br] G = graphviz.Digraph()[br] lst=sorted(self.edges)[br] for edge in lst:[br] depth,source,destination,lbl = edge[br] G.edge(str(source), str(destination), label = lbl)[br] return G[br][/b][/color] [br]# ここからが実行[br]myBST=BST()[br]nums = [20,30,40,5,10,15,25,35][br]myBST.make_from(nums)[br][b]myBST.insert(5)[br]myBST.insert(50)[br]myBST.show_graph()[br][/b]#=====================[br][OUT][br]データ5が重複しています。
[size=150][size=100]# その他のメソッド (削除、最小値、最大値など)[br]myBST=BST()[br]nums = [20,30,40,5,10,15,25,35][br]myBST.make_from(nums)[br]myBST.insert(50)[br]myBST.insert(60)[br]myBST.show_graph()[br]のように、どんどん、大きい数を追加すると、木はrの枝が続く。アンバランスだ。[br][br]また、[br]# その他のメソッド (削除、最小値、最大値など)[br]myBST=BST()[br]nums = [20,30,40,5,10,15,25,35][br]myBST.make_from(nums)[br]myBST.insert(32)[br]myBST.insert(31)[br]myBST.show_graph()[br]のように、少しずつ小さく数を順に追加すると、木はLの枝が続く。いっそうアンバランスだ。[br][/size][/size]
[b][size=150]<データの活用>[/size][/b][br][br]BSTを作っているからには、データ追加だけではなく、[br]最大値、最小値を求めたりもしたいですね。[br][br][b][color=#9900ff][u]質問:最大値や最小値を得る関数はどうやれば作れるでしょうか。[br][/u][/color][/b][br][size=85][size=100]BSTを作っているからには、データ追加だけではなく、[br]最大値、最小値を求めたりもしたいですね。[br][/size][size=100]class BSTのメソッド関数として[br]次の2つを追加して使えばよいね。[br]根から左に進み続けて、もう左に行けなくなったときの点の値が最小値[br]根から右に進み続けて、もう右に行けなくなったときの点の値だ最大値だ。[br]この考え方をコードにしてみよう。[br][br] def min(self):[br] return self._min_recursive(self.root,None)[br] def _min_recursive(self, v, previous):[br] if v.left:[br] previous = v[br] return self._min_recursive(v.left, previous)[br] else:[br] return v.data[br] def max(self):[br] return self._max_recursive(self.root,None)[br] def _max_recursive(self, v, previous):[br] if v.right:[br] previous = v[br] return self._max_recursive(v.right, previous)[br] else:[br] return v.data[br][br][/size][/size]# ここからが実行[br]myBST=BST()[br]nums = [20,30,40,5,10,15,25,35][br]myBST.make_from(nums)[br]myBST.insert(50)[br]print(f"min = {myBST.min()}")[br]print(f"max = {myBST.max()}")[br]myBST.show_graph()[br][br]#=======================[br][OUT][br]min = 5[br]max = 50[br][br]データの最大、最小を求めるには、別に2分探索木を使わなくても、配列やリストでできます。[br]データ構造の選択やその実装のコードを改良するということは、[br]大量のデータを扱う必要がある状況では処理時間を削減することにつながるので、大変価値がある。[br]あるいは、処理時間を競うコンテストでもそうでしょう。[br][br]関心のある人は、さらにコードの改良をめざしてみよう。

Information: Bstをclassで作ろう