非递归实现BST树的删除操作

非递归实现BST树的删除操作:

     1. 先搜索BST树,找到待删除的节点
     2. 判断删除节点是否有两个孩子,如果有,用前驱的值代替,直接删除前驱
     3. 删除有一个孩子的节点,或者没有孩子的节点(看作有一个孩子,孩子是null)
/**
 * 非递归实现BST树的删除操作
 * @param data
 */
public void non_remove(T data){
    if(this.root == null){
        return;
    }
    // 1. 先搜索BST树,找到待删除的节点
    BSTNode cur = this.root;
    BSTNode parent = null;
    while(cur != null){
        if(cur.getData().compareTo(data) > 0){
            parent = cur;
            cur = cur.getLeft();
        } else if(cur.getData().compareTo(data) < 0){
            parent = cur;
            cur = cur.getRight();
        } else {
            break;
        }
    }

    if(cur == null){ // 表示BST树中没有值为data的节点
        return;
    }
    // 2. 判断删除节点是否有两个孩子,如果有,用前驱的值代替,直接删除前驱
    if(cur.getLeft() != null && cur.getRight() != null){
        // 找前驱节点,用前驱节点的值代替待删节点的值,然后直接删除前驱节点cur
        BSTNode old = cur;
        parent = cur;
        cur = cur.getLeft();
        while(cur.getRight() != null){
            parent = cur;
            cur = cur.getRight();
        }
        old.setData(cur.getData()); // 前驱节点的值代替待删节点的值
    }

    // 3. 删除有一个孩子的节点,或者没有孩子的节点(看作有一个孩子,孩子是null)
    BSTNode child = cur.getLeft();
    if(child == null){
        child = cur.getRight();
    }

    if(parent == null){ // 删除的就是根节点
        this.root = child;
    } else {
        if(parent.getLeft() == cur){
            parent.setLeft(child);
        } else {
            parent.setRight(child);
        }
    }
}

你可能感兴趣的:(非递归实现BST树的删除操作)