LeetCode 701: 二叉搜索树中的插入操作
题目描述
给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 保证原始二叉搜索树中不存在新值。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。
例如,
给定二叉搜索树:
4
/ \
2 7
/ \
1 3
和 插入的值: 5
你可以返回这个二叉搜索树:
4
/ \
2 7
/ \ /
1 3 5
或者这个树也是有效的:
5
/ \
2 7
/ \
1 3
\
4
解题
最简单的思路和LeetCode 700: Search in a Binary Search Tree类似,如果当前节点值比待插入值大,那么这个值将插入到该节点的左子树,即应该继续与左子节点比较,如果当前节点值小于带插入值,则插入到右子树,即应该继续与右子节点比较,直到下一个要比较的节点为空时,将该值插入到空节点为止即可。这种方法优点是容易实现,思路简单,缺点是插入节点较多时可能会导致树严重不平衡,增加搜索、比较、插入等的消耗。
循环法
/**
* 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* insertIntoBST(TreeNode* root, int val) {
if (root==NULL){
root = new TreeNode(val);
return root;
}
TreeNode *cur_node = root;
while (cur_node){
if (cur_node->val > val){
if (cur_node->left)
cur_node = cur_node->left;
else{
cur_node->left = new TreeNode(val);
break;
}
}
else{
if (cur_node->right)
cur_node = cur_node->right;
else{
cur_node->right = new TreeNode(val);
break;
}
}
}
return root;
}
};
迭代法
/**
* 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* insertIntoBST(TreeNode* root, int val) {
if (root==NULL){
return new TreeNode(val);
}
if (root->val > val)
root->left = insertIntoBST(root->left, val);
else
root->right = insertIntoBST(root->right, val);
return root;
}
};