cs: add “450. Delete Node in a BST”
Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
parent
c838c99927
commit
fe58c04720
1 changed files with 45 additions and 0 deletions
45
cs/delete-node-in-a-bst.cs
Normal file
45
cs/delete-node-in-a-bst.cs
Normal 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;
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue