在二叉查找树中插入节点

来源:互联网 发布:淘宝海外 编辑:程序博客网 时间:2024/06/15 13:16

题目:

给定一棵二叉查找树和一个新的树节点,将节点插入到树中。

你需要保证该树仍然是一棵二叉查找树。

 注意事项

You can assume there is no duplicate values in this tree + node.

样例

给出如下一棵二叉查找树,在插入节点6之后这棵二叉查找树可以是这样的:

  2             2 / \           / \1   4   -->   1   4   /             / \   3             3   6
思路:

采用递归法,一直找到叶子结点进行插入即可。

代码:

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root: The root of the binary search tree.
     * @param node: insert this node into the binary search tree
     * @return: The root of the new binary search tree.
     */
    TreeNode* insertNode(TreeNode* root, TreeNode* node) {
        // write your code here
        if(root==NULL) return node;
        int k=node->val;
        if(root->val>k)
           {  if(root->left!=NULL)
                  { insertNode(root->left,node);
                  }
               else root->left=node;
               return root;
           }
        if(root->val<k)
           {  if(root->right!=NULL)
                  { insertNode(root->right,node);
                  }
               else root->right=node;
               return root;
           }
    }
};

感想:这道题没什么难的,就是我做完后一直不对,我就比着又重新打了一遍就对了,可能中间有中文字符,这也说明有时候题目过不了不一定是代码的理论问题。

原创粉丝点击