JS-常用的数据结构之树、二叉树、平衡二叉树

二叉树的实现代码

  1 // 二叉树构造
  2 function BinaryTree(){
  3     var Node=function(key){         //节点函数
  4         this.key=key;
  5         this.left=null;
  6         this.right=null;
  7     }
  8 
  9     var root = null;
 10     this.insert = function(key){
 11         var newNode = new Node(key);
 12         if(root == null){
 13             root = newNode;
 14         }else{
 15             insertNode(root, newNode);
 16         }
 17     }
 18 
 19     var insertNode = function(node, newNode){
 20         if(node.key > newNode.key){
 21             if(node.left == null){
 22                 node.left = newNode;
 23             }else{
 24                 insertNode(node.left, newNode);
 25             }
 26         }else{
 27             if(node.right == null){
 28                 node.right = newNode;
 29             }else{
 30                 insertNode(node.right, newNode);
 31             }
 32         }
 33     }
 34 
 35     var inOrderTraverseNode = function(node, callback){
 36         if(node != null){
 37             inOrderTraverseNode(node.left, callback);
 38             callback(node.key);
 39             inOrderTraverseNode(node.right, callback);
 40         }
 41     }
 42     // 中序遍历 
 43     this.inOrderTraverse = function(callback){
 44         inOrderTraverseNode(root, callback);
 45     }
 46 
 47     var proOrderTraverseNode=function(node,callback){
 48         if(node!=null){
 49             callback(node.key)
 50             proOrderTraverseNode(node.left,callback)
 51             proOrderTraverseNode(node.right,callback)
 52         }
 53     }
 54     //前序遍历  复制   效率高得多
 55     this.proOrderTraverse=function(callback){
 56         proOrderTraverseNode(root,callback);
 57     }
 58 
 59     var postOrderTraverseNode=function(node,callback){
 60         if(node!=null){
 61             postOrderTraverseNode(node.left,callback)
 62             postOrderTraverseNode(node.right,callback)
 63             callback(node.key)
 64         }
 65     }
 66     //后续遍历  文件系统遍历
 67     this.postOrderTraverse=function(callback){
 68         postOrderTraverseNode(root,callback);
 69     }
 70 
 71     // 查找
 72     // 查找最大值
 73     var maxNode=function(node){
 74         if(node){
 75             while(node&&node.right!=null){
 76                 node=node.right;
 77             }
 78             return node.key;
 79         }
 80     }
 81     this.max=function(){
 82         return maxNode(root);
 83     }
 84 
 85     // 查找最小值
 86     var minNode = function(node){
 87         if (node){
 88             while(node && node.left != null){
 89                 node = node.left;
 90             }
 91             return node.key;
 92         }
 93     }    
 94     this.min=function(){
 95         return minNode(root);
 96     }
 97 
 98     // 查找指定值
 99     var searchNode=function(node,key){
100         if (node == null){
101             return false;
102         }
103 
104         if(node.key > key){
105             searchNode(node.left, key);
106         }else if (node.key < key){
107             searchNode(node.right, key);
108         }else{
109             return true;
110         }
111     }
112     this.search=function(key){
113         return searchNode(root,key)
114     }
115 
116    /**
117     * 二叉树的删除
118     * 二叉树删除分三种情况
119     * 删除左叶子节点
120     * 删除右叶子节点
121     * 以及拥有左右两个节点的主节点删除
122     * 拥有左右两个节点的主节点删除要考虑到数据的可排序行需要将删掉的节点重新赋值
123     */
124    var findMinNode=function(node){ //如果存在左右两个节点的话查找右节点的最小节点
125         if(node){
126             while(node&&node.left!=null){
127                 node=node.left
128             }
129             return node;
130         }
131         return null
132     }
133 
134    var removeNode=function(node,key){
135        if (node == null){
136             return null;
137        }
138 
139        if(node.key > key){ // 递归查找左叶子节点，直接等于返回的null值
140             node.left = removeNode(node.left, key);
141             return node;
142        }else if(node.key < key){
143             node.right = removeNode(node.right, key);
144             return node;
145        }else{
146             if(node.left==null&&node.right==null){// 当只有一个节点，而且被选中
147                 node=null;
148                 return node;
149             }
150             if(node.left==null){ //左节点为空
151                 node=node.right;
152                 return node;            
153             }else if(node.right==null){//右节点为空
154                 node=node.left;
155                 return node;
156             }
157             
158             // ?????
159             var aux=findMinNode(node.right);//查找到右节点最小节点赋值
160             node.key=aux.key;
161             node.right=removeNode(node.right,aux.key);
162             return node;
163        }
164    }
165    this.remove=function(key){
166         return removeNode(root,key)
167     }
168 }
169 
170 // test
171 var nodes=[8,3,10,2,9,14,4,7,13];
172 var binaryTree=new BinaryTree();
173 nodes.forEach(function(key){
174     binaryTree.insert(key);
175 })
176 var callback=function(key){
177     document.write(key);
178 }
179 binaryTree.inOrderTraverse(callback);
180 
181 document.write("+++++++++" + binaryTree.max());