leetcode450.删除二叉搜索树中的节点

题目大意

给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。

一般来说，删除节点可分为两个步骤：

首先找到需要删除的节点；
如果找到了，删除它。
说明： 要求算法时间复杂度为 O(h)，h 为树的高度。

示例:

root = [5,3,6,2,4,null,7]
key = 3

    5
   / \
  3   6
 / \   \
2   4   7

给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。

一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。

    5
   / \
  4   6
 /     \
2       7

另一个正确答案是 [5,2,6,null,4,null,7]。

    5
   / \
  2   6
   \   \
    4   7

解题思路

找到该节点后，如果左子树为空，返回右子树；如果右子树为空，返回左子树；如果都不为空：在右子树中找到最左节点，将左子树挂到该节点下，返回右子树；
如果节点值大于key，在左子树中重复上述操作；
如果节点值小于key，在右子树中重复上述操作；

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
    	if (root == nullptr)
    		return nullptr;

    	if (root->val == key){
    		if(!root->left)
    			return root->right;
    		else if (!root->right)
    			return root->left;
    		else{
    			TreeNode * tmp = root->right;
    			while (tmp->left)
    				tmp = tmp->left;
    			tmp->left = root->left;
    			return root->right;
    		}
    	}
    	else if (root->val > key)
    		root->left = deleteNode(root->left, key);
    	else
    		root->right = deleteNode(root->right, key);
    	return root;
    }
};