BST:노드 제거

1456 단어 두 갈래 나무
BST와 키를 지정합니다. 값이 키인 노드를 찾고 삭제해야 합니다. 삭제된 두 갈래 트리인지 BST인지 확인하십시오.
방법1: 이 노드를 찾아서 이 노드의 오른쪽 노드를 대체하고 이 노드의 왼쪽 트리를 오른쪽 노드의 맨 왼쪽 아래 노드의 left에 연결합니다.
    public TreeNode deleteNode(TreeNode root, int key) {
        if(root == null)
            return null;
        TreeNode node = root;
        TreeNode pre = null;
        boolean isLeft = true;
        while(node != null && node.val != key){ // key
            pre = node;
            if(key < node.val){
                node = node.left;
                isLeft = true;
            }
            else{
                node = node.right;
                isLeft = false;
            } 
        }
        if(node == null) // , 
            return root;
        if(node.right == null){ // null,  
            if(pre == null) return node.left;
            else{
                if(isLeft) pre.left = node.left;
                else pre.right = node.left;
                return root;
            }
        }  
        else{  //node.right != null
            TreeNode l = node.left;
            TreeNode r = node.right;
            node = node.right;
            while(node.left != null){ // 
                node = node.left;
            }
            node.left = l;
            if(pre == null)
                return r;
            else{
                if(isLeft) pre.left = r;
                else pre.right = r;
                return root;
            }
        }
    }

좋은 웹페이지 즐겨찾기