数据结构 - 二叉树，AVL树

学习目标

1.二叉树
2.AVL树

树与二叉树

树是一种数据结构
比如:目录结构
树是一种可以递归定义的数据结构
树是由n个节点组成的集合:
如果n=0，那这是一棵空树;
如果n>0，那存在1个节点作为树的根节点，其他节点可以分为m个集合，每个集合本身又是一棵树。

树的实例：模拟文件系统

class Node:
	def __init__(self, name, type='dir'):
		self.name = name
		self.type = type
		self.children = []
		self.parent = None
	
	def __repr__(self):
		return self.name

class FileSystemTree:
	def __init__(self):
		self.root = Node("/")
		self.now = self.root
	
	def mkdir(self, name):
		if name[-1] != "/":
			name += "/"
		node = Node(name)
		self.now.children.append(node)
		node.parent = self.now
	
	def ls(self):
		return self.now.children
	
	def cd(self, name):
		if name[-1] != "/":
			name += "/"
		if name == "../":
			self.now = self.now.parent
			return
		for child in self.now.children:
			if child.name == name:
				self.now = child
				return
		raise ValueError("invalid dir")

tree = FileSystemTree()
tree.mkdir("var/")
tree.mkdir("var2/")
tree.mkdir("var1/")
tree.cd("var/")
tree.mkdir("python")
print(tree.ls())
	

二叉树

叉树的链式存储:将二叉树的节点定义为一个对象，节点之间通过类似链表的链接方式来连接。

class BiTreeNode:
	def __init__(self, data):
		self.data = data
		self.lchild = None
		self.rchild = None
a = BiTreeNode("A")
b = BiTreeNode("B")
c = BiTreeNode("C")
d = BiTreeNode("D")
e = BiTreeNode("E")
f = BiTreeNode("F")
g = BiTreeNode("G")

e.lchild = a
e.rchild = g
a.rchild = c
c.lchild = b
c.rchild = d
g.rchild = f

root = e
print(root.lchild.rchild.data)

二叉树的遍历

二叉树的遍历方式:
前序遍历: EACBDGF
中序遍历: ABCDEGF
后序遍历: BDCAFGE
层次遍历: EAGCFBD

def pre_order(root):
	if root:
		print(root.data, end=',')
		pre_order(root.lchild)
		pre_order(root.rchild)

def in_order(root):
	if root:
		in_order(root.lchild)
		print(root,data, end=',')
		in_order(root.rchild)