lintcode 85. 在二叉查找树中插入节点

MissZhou要努力

于 2019-09-22 00:27:15 发布

阅读量268

点赞数

分类专栏： leetcode 文章标签： leetcode 二叉树

本文链接：https://blog.youkuaiyun.com/zhou_yujia/article/details/101128661

版权

leetcode 专栏收录该内容

89 篇文章

订阅专栏

描述

中文English

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

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

保证不会出现重复的值

您在真实的面试中是否遇到过这个题？是

题目纠错

样例

样例  1:
	输入: tree = {}, node= 1
	输出: {1}
	
	样例解释:
	在空树中插入一个点，应该插入为根节点。

	
样例 2:
	输入: tree = {2,1,4,3}, node = 6
	输出: {2,1,4,3,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.
     */
    void dfs(TreeNode * root, TreeNode * node) {
        if (root == NULL) {
            root = node;
            return;
        }
        if (root->val > node -> val) {
            if (root->left)
                dfs(root->left, node);
            else {
                root->left = node;
                return;
            }
        } else {
            if (root->right)
                dfs(root->right, node);
            else {
                root->right = node;
                return;
            }
        }
    }
    TreeNode * insertNode(TreeNode * root, TreeNode * node) {
        // write your code here
        if(root) {
            dfs(root, node);
            return root;
        } else {
            return node;
        }
         
    }
};