了解了二叉堆之后,二叉搜索树就好说了,就是一个节点,左边的子节点是不可能比他大的,右边的子节点是一定大于它的,想了半天终于把创建给写好了。
创建
- import UIKit
-
- var str = "二叉搜索树"
- //这个就不跟前面的完全二叉树一样了,得自己建了类或者结构体了,我建了个类
- class erchaTreeNote {
- var data: Int
- var leftChild: erchaTreeNote!
- var rightChild: erchaTreeNote!
- init(data:Int) {
- self.data = data
- }
- }
-
- var a = [12,321,432,213,423,4]
-
- func createTree() -> (erchaTreeNote) {
-
- let root = erchaTreeNote(data:a[0]);
- for x in a[1...a.count-1] {
-
- let child = erchaTreeNote(data: x)
-
- var temp = root
- //循环的条件想了半天,想着如何能走下去,在纸上练了几遍,发现了规律,本来进来一个数如果它加进去了,它的左右子节点都是空的,再往下就不走了,但是这个是走不通的,再想他们有什么共性,我就想,既然把它按在了树上,那它再走一次,必然和上一次的路径是一样的,当我找到和它一模一样的时候,就是结束的时候,如果我找不到它,一直都不能结束。就按这个条件走就出来了。
-
- while temp !== child {
- //如果进来的数小于父节点
- if child.data < temp.data {
- //不为空,那我就把父节点左边子节点拿上,再重新来过
- if temp.leftChild != nil
- {
- temp = temp.leftChild
- }
- //当父节点左边是空的时候,那就直接填上
- else
- {
- temp.leftChild = child
- print("\(temp.data)左边的孩子")
- temp = child //优化语句
- }
- }else //进来的数大于父节点
- {
- //不为空,那我就把父节点右边子节点拿上,重新来过
- if temp.rightChild != nil {
- temp = temp.rightChild
- }
- //当父节点的右边为空的时候,那就直接补上
- else
- {
- temp.rightChild = child
- print("\(temp.data)右边的孩子")
- temp = child //优化语句
- }
- }
- }
- print(child.data)
- }
- return root
- }
- createTree()
最大值和最小值
- //找到最大的,那就直接朝右走下去,最小则是朝左走下去
- func FindMax(_ shu:erchaTreeNote) -> (erchaTreeNote){
- var temp = shu
- //如果右边不为空,一直找下去
- while temp.rightChild != nil {
- temp = temp.rightChild
- }
- return temp
- }
- //最小
- func FindMin(_ shu:erchaTreeNote) -> (erchaTreeNote){
- var temp = shu
- //如果左边不为空,一直找下去
- while temp.leftChild != nil {
- temp = temp.leftChild
- }
- return temp
- }
查找
- //找到特定的那个数,这个我发现如果两个一样的数字,可能进入树的时间不一样,所以他们位置也就不一样,得把这棵树可能的条件都循环完毕才可以
- func Findfrom(_ shu:erchaTreeNote,_ mubiao:Int) -> ( Array<erchaTreeNote>){
- var temp = shu
-
- var arr = Array<erchaTreeNote>()
-
- while true{
- if temp.data < mubiao
- {
- if temp.rightChild != nil
- {
- print("拐进右边\(temp.rightChild.data)")
- temp = temp.rightChild
- }
- else
- {
- print("没有啦")
- break
- }
- }
- else if temp.data > mubiao
- {
- if temp.leftChild != nil
- {
- print("拐进左边\(temp.leftChild.data)")
- temp = temp.leftChild
- }
- else
- {
- print("没有啦")
- break
- }
-
- }
- else
- {
- print("找到啦它就是\(temp.data)")
- //找到它的时候给它加载数组里,不确定还有没有一样的存在
- arr.append(temp)
- if (temp.rightChild != nil)
- {
- print("下面还有东西呢\(temp.rightChild.data)")
- temp = temp.rightChild
- }
- else
- {
- break
- }
- }
- }
- return arr
- }
插入
- //插入,其实就跟创建是一模一样,只不过多了一个数而已
- func insert(_ shu:erchaTreeNote,_ x :Int) -> (erchaTreeNote)
- {
- func digui (_ ee: erchaTreeNote)
- {
- if ee.data > x {
- if ee.leftChild != nil
- {
- digui(ee.leftChild)
- }
- else
- {
- print("它的父节点是\(ee.data).他在左边")
- ee.leftChild = erchaTreeNote(data: x)
- }
- }
- else
- {
- if ee.rightChild != nil
- {
- digui(ee.rightChild)
- }
- else
- {
- print("它的父节点是\(ee.data).他在右边")
- ee.rightChild = erchaTreeNote(data: x)
- }
- }
- }
-
- digui(shu)
- return shu
- }
删除
删除好做,但是得找到那个能顶替它原来位置的节点,我这里只是打印出来,因为没有父节点,不好去找,所以就没做。。
- //移除的逻辑也简单易懂,删除这个节点,如果有右节点,再去找右边最小的那个顶上,如果没有右节点,左节点顶上,要是都木有,那就删了
-
- func yichu(_ shu:erchaTreeNote,_ x:Int)
- {
- //先找到这个节点,因为里面可能有重复的情况发生,所以得删个几次,我们从最深的那个删起
- let arr = Findfrom(shu,x)
- arr.count
- for i in 0..<arr.count
- {
- let temp = arr[arr.count - i - 1]
-
- if temp.rightChild != nil
- {
- print("删了\(temp.data),与\(FindMin(temp.rightChild).data)发生调换")
- }
- else if temp.leftChild != nil
- {
- FindMax(temp.rightChild)
- print("删了\(temp.data),与\(FindMax(temp.rightChild).data)发生调换")
- }
- else
- {
- print("删了\(temp.data),没有发生调换")
- }
- }
- }
前驱
- //前驱,一个节点的前驱是指所有比它小的节点里面最大的那个;
- func Getpredecessor(_ shu:erchaTreeNote,_ x :Int) -> (erchaTreeNote)
- {
- let arr = Findfrom(shu,x)
- for temp in arr
- {
- if temp.leftChild != nil
- {
- return FindMin(temp.leftChild)
- }
- else
- {
- return temp
- }
- }
- print("\(x)并没有找到")
- return erchaTreeNote(data: -1)
- }
后继
- //后继,一个节点的后继是指所有比它大的节点里面最小的那个
- func GetSuccessor(_ shu:erchaTreeNote,x)
- for temp in arr
- {
- if temp.rightChild != nil
- {
- return FindMax(temp.rightChild)
- }
- else
- {
- return temp
- }
- }
- print("\(x)并没有找到")
- return erchaTreeNote(data: -1)
- }
中序遍历
- //遍历 Inorder Traversal 中序遍历,中序遍历就是先左后右,很直观的输出数字。
- func InorderTra(_ shu:erchaTreeNote){
-
- var arr = Array<erchaTreeNote>()
- func lunhui(_ shu:erchaTreeNote)
- {
- if shu.leftChild != nil
- {
- arr.append(shu)
- lunhui(shu.leftChild)
- }
- else
- {
- print(shu.data)
- let beis = arr.remove(at: arr.count-1)
- print(beis.data)
- if(beis.rightChild != nil) {
- lunhui(beis.rightChild)
- }
- }
- }
- lunhui(shu)
-
- }
就酱,还是蛮有成就感的。要是不对,咱们一起讨论,当然里面的一些极端情况我没有做判断,只是想着熟悉下思路。