cs: add “450. Delete Node in a BST”

Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
Matej Focko 2024-01-09 12:25:16 +01:00
parent c838c99927
commit fe58c04720
Signed by: mfocko
GPG key ID: 7C47D46246790496

View file

@ -0,0 +1,45 @@
/**
* Definition for a binary tree node.
* public class TreeNode {
* public int val;
* public TreeNode left;
* public TreeNode right;
* public TreeNode(int val=0, TreeNode left=null, TreeNode right=null) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
public class Solution {
public TreeNode DeleteNode(TreeNode root, int key) {
if (root == null) {
return null;
}
if (key == root.val) {
if (root.left == null) {
return root.right;
}
if (root.right == null) {
return root.left;
}
// find successor
var node = root.right;
while (node.left != null) {
node = node.left;
}
root.val = node.val;
root.right = DeleteNode(root.right, node.val);
} else if (key < root.val) {
root.left = DeleteNode(root.left, key);
} else if (key > root.val) {
root.right = DeleteNode(root.right, key);
}
return root;
}
}