https://leetcode-cn.com/problems/delete-node-in-a-bst/
题目
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
- 首先找到需要删除的节点;
- 如果找到了,删除它。
- 说明: 要求算法时间复杂度为 O(h),h 为树的高度。
示例
思路
用递归的方法
- 如果当前节点不是要删除的节点,根据二叉搜索树的性质往孩子节点去找
- 如果是,分为几种情况
(1) 当前节点没有孩子,直接delete, return NULL即可
(2)当前节点仅有一边孩子,那么将将root的孩子赋给root,再删掉root即可
(3) 有两边孩子,那么我们有两种解决方案,因为为了保持原有性质,root -> val要大于左边所有,小于右边所有,那么我们可以找到左边的最大值或者右边的最小值,将root -> val赋值成这个值,再根据你是在哪一边找的,递归地删掉这个值即可
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int getNew(TreeNode* root) {
for (; root -> left; root = root -> left) ;
return root -> val;
}
TreeNode* deleteNode(TreeNode* root, int key) {
if (!root) return root;
if (root -> val < key) root -> right = deleteNode(root -> right, key);
if (root -> val > key) root -> left = deleteNode(root -> left, key);
if (root -> val == key) {
if (!root -> left && !root -> right) {
delete root;
return NULL;
}
if (!(root -> left)) {
TreeNode* p = root -> right;
delete root; return p;
}
if (!(root -> right)) {
TreeNode* p = root -> left;
delete root; return p;
}
if (root -> left && root -> right) {
int k = getNew(root -> right);
root -> val = k;
root -> right = deleteNode(root -> right, k);
}
}
return root;
}
};