在上一節我們講述了AVL樹的基本知識,並實現了自平衡二叉尋找樹的插入操作,這一節裡我們來實現一下刪除操作。
刪除操作的Java代碼實現如下所示,其他代碼和前一講相同。
//刪除一個節點public void remove(int data){rootnode=remove(data,this.rootnode);}private AVLTreeNode remove(int data,AVLTreeNode root){if(root==null){return root;}if(data<root.data){root.left=remove(data,root.left);}else if(data>root.data){root.right=remove(data,root.right);}else if(root.left!=null&&root.right!=null){root.data=minAVLTreeNode(root.right).data;root.right=remove(root.data,root.right);}else{root=root.left!=null?root.left:root.right;}return balance(root);}//擷取以begin為根節點的樹的最小關鍵字節點的引用private AVLTreeNode minAVLTreeNode(AVLTreeNode begin){if(begin==null)return null;while(begin.left!=null){begin=begin.left;}return begin;}
在主程式中我們做如下處理:
mytree.insert(4);mytree.insert(3);mytree.insert(2);mytree.insert(1);mytree.insert(5);mytree.remove(4);mytree.remove(5);
運行結果如下所示:
遞迴方式實現前序走訪:2 1 3 遞迴方式實現中序遍曆:1 2 3 遞迴方式實現後序遍曆:1 3 2 非遞迴方式實現前序走訪:2 1 3 非遞迴方式實現中序遍曆:1 2 3 非遞迴方式實現後序遍曆:1 3 2
從運行結果中我們可以看出,在刪除之後,剩下的1,2,3三個節點已經不滿足平衡條件,同過平衡化處理之後得到一個新的平衡化二叉尋找樹。