数据结构与算法——树的广度优先遍历
生活随笔
收集整理的這篇文章主要介紹了
数据结构与算法——树的广度优先遍历
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
實現:往完全二叉樹中添加一個節點,使得添加之后這棵樹依舊是一棵完全二叉樹
class Node:def __init__(self, val):self.val = val # 數據域self.left = None # 左指針域self.right = None # 右指針域class Tree:def __init__(self):self.root = None # 樹的根節點def add(self, val):# 層次遍歷, 廣度優先遍歷# val 要添加的節點的值# 往樹中添加一個節點,并保證添加之后這棵樹依舊是一棵完全二叉樹node = Node(val)# 判斷樹是否為空,如果為空,直接將node設置為根節點if not self.root:self.root = nodereturn# 從上往下,從左往右的去遍歷整棵樹,然后找到第一個空位# 把節點添加進去queue = [self.root] # 存每一層的節點while True:# 第一次 queue = [root]# 第二次 queue = [root.left, root.right]# queue = [root.right, root.left.left, root.left.right]# queue = [root.left.left, root.left.right, root.right.left, root.right.right]cur_node = queue.pop(0)# 先找左邊,看有沒有空位if not cur_node.left:cur_node.left = nodereturn# 左邊沒有空位,就找右邊elif not cur_node.right:cur_node.right = nodereturn# 如果都沒有空位,那就把左邊節點與右邊節點都加到之后要判斷的節點中queue.extend((cur_node.left, cur_node.right))def show(self):# 展示樹if not self.root:return# 從上往下,從左往右的去遍歷整棵樹,然后找到第一個空位# 把節點添加進去queue = [self.root] # 存每一層的節點i = 1while queue:size = len(queue) # 當前層的元素個數print(f'第{i}層', end='\t')for _ in range(size):node = queue.pop(0) # 隊列中的第一個元素拋出來print(node.val, end=' ') # 對當前元素進行操作# 節點的左孩子與右孩子添加到隊列里if node.left:queue.append(node.left)if node.right:queue.append(node.right)print()i += 1 if __name__ == '__main__':tree = Tree()tree.add(0)tree.add(1)tree.add(2)tree.add(3)tree.add(4)tree.add(5)tree.add(6)tree.add(7)tree.show()執行結果:
總結
以上是生活随笔為你收集整理的数据结构与算法——树的广度优先遍历的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 数据结构与算法——二叉树的深度优先遍历
- 下一篇: shell脚本之变量的作用域