1.题目要求:
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
1.首先找到需要删除的节点;
2.如果找到了,删除它。
2.题目代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
//如果结点为空或者删除的是叶子结点则返回NULL
if(root == NULL || (root->left == NULL && root->right == NULL && root->val == key)){
return NULL;
}
//如果要删除的结点左不为空,右为空,则返回结点的左节点
if(root->left != NULL && root->right == NULL && root->val == key){
return root->left;
}
//如果要删除的结点右不为空,左为空,则返回结点的右节点
if(root->left == NULL && root->right != NULL && root->val == key){
return root->right;
}
//如果删除的结点右不为空,左不为空,则删除结点的 左子树要移到右子树最左边
if(root->left != NULL && root->right != NULL && root->val == key){
TreeNode* right_node = root->right;
TreeNode* cur = right_node;
while(cur->left != NULL){
cur = cur->left;
}
cur->left = root->left;
return root->right;
}
if(root->val > key){
root->left = deleteNode(root->left,key);
}
if(root->val < key){
root->right = deleteNode(root->right,key);
}
return root;
}
};