450. Delete Node in a BST

最新推荐文章于 2021-03-30 11:07:04 发布

云养猫v2

最新推荐文章于 2021-03-30 11:07:04 发布

阅读量153

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/weixin_39590058/article/details/106896514

leetcode 专栏收录该内容

216 篇文章

订阅专栏

本文详细介绍了如何在二叉搜索树中删除指定值的节点，包括三种情况：删除叶子节点、删除有一个子节点的节点和删除有两个子节点的节点。通过递归调用和寻找后继节点或前驱节点来保持树的性质。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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;
}