450. Delete Node in a BST

public TreeNode deleteNode(TreeNode root, int key) {
	if (root == null) return null;

	if      (key < root.val) root.left  = deleteNode(root.left, key);
	else if (key > root.val) root.right = deleteNode(root.right, key);
	else {
		if 		(root.left  == null) return root.right;
		else if (root.right == null) return root.left;

		// TreeNode minNode = sucessor(root);
		root.val = successorVal(root);
		root.right = deleteNode(root.right, root.val);
	}
	return root;
}

public int successorVal(TreeNode root) {
	root = root.right;
	while (root.left != null) 
		root = root.left;
	return root.val;
}


public int successor(TreeNode root) {
	root = root.right;
	while (root.left != null) 
		root = root.left;
	return root.val;
}

public int predecessor(TreeNode root) {
	root = root.left;
	while (root.right != null)
		root = root.right;
	return root.val;
}

public TreeNode deleteNode(TreeNode root, int key) {
	if (root == null) return null;
	
	if      (key > root.val) root.right = deleteNode(root.right, key);
	else if (key < root.val) root.left  = deleteNode(root.left, key);
	else {
		if (root.left == null && root.right == null) root = null;
		else if (root.right != null) {
			root.val = successor(root);
			root.right = deleteNode(root.right, root.val);
		} else {
			root.val = predecessor(root);
			root.left = deleteNode(root.left, root.val);
		}
	}
	return root;
}
以下是一个二叉搜索树(Binary Search Tree)的 Python 实现,包括了插入、删除、搜索和转换为列表等方法。 ```python class TreeNode: """The tree node class""" def __init__(self, val): self.val = val self.left = None self.right = None class BSTree: """The binary search tree""" def __init__(self, val): """Initialize the BSTree""" self.root = TreeNode(val) def search(self, val): """Search for a value in the tree, and return the tree node @return: the tree node that contain the val None if val is not in the tree """ curr_node = self.root while curr_node: if curr_node.val == val: return curr_node elif curr_node.val > val: curr_node = curr_node.left else: curr_node = curr_node.right return None def insert(self, val): """Insert a value""" new_node = TreeNode(val) if not self.root: self.root = new_node return curr_node = self.root while curr_node: if curr_node.val > val: if curr_node.left: curr_node = curr_node.left else: curr_node.left = new_node return else: if curr_node.right: curr_node = curr_node.right else: curr_node.right = new_node return def delete(self, val): """Delete a value from the tree""" def find_min_node(node): while node.left: node = node.left return node def delete_node(node, val): if not node: return None if node.val == val: if not node.left and not node.right: return None if not node.left: return node.right if not node.right: return node.left min_node = find_min_node(node.right) node.val = min_node.val node.right = delete_node(node.right, min_node.val) elif node.val > val: node.left = delete_node(node.left, val) else: node.right = delete_node(node.right, val) return node self.root = delete_node(self.root, val) def toList(self): """Convert tree values into a list in in-order traversal""" res = [] def inorder_traversal(node): if not node: return inorder_traversal(node.left) res.append(node.val) inorder_traversal(node.right) inorder_traversal(self.root) return res ``` 使用方法: ```python bst = BSTree(5) bst.insert(3) bst.insert(7) bst.insert(2) bst.insert(4) bst.insert(6) bst.insert(8) print(bst.toList()) # [2, 3, 4, 5, 6, 7, 8] bst.delete(5) print(bst.toList()) # [2, 3, 4, 6, 7, 8] node = bst.search(7) print(node.val) # 7 ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值