新手算法學習之路----二叉樹(在一個二叉查找樹中插入一個節點)

題目:給定一棵二叉查找樹和一個新的樹節點,將節點插入到樹中。node

          你須要保證該樹仍然是一棵二叉查找樹。this

給出以下一棵二叉查找樹,在插入節點6以後這棵二叉查找樹能夠是這樣的:spa

須要搞清楚定義:二叉排序樹或者是一棵空樹;或者是具備下列性質的二叉樹:
(1)若左子樹不空,則左子樹上全部結點的值均小於它的根結點的值;
(2)若右子樹不空,則右子樹上全部結點的值均大於它的根結點的值;
(3)左、右子樹也分別爲二叉排序樹;

Java代碼:2 2 / \ / \ 1 4 --> 1 4 / / \ 3 3 6
/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */
public class Solution {
    /**
     * @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.
     */
    public TreeNode insertNode(TreeNode root, TreeNode node) {
        // write your code here
        if(root==null){
            return node;
        }
        if(root.val>node.val){                       //這個樹裏面沒有重複的數,因此無需考慮root.val == node.val的狀況
            root.left = insertNode(root.left, node);  //待插入值確定在左右子樹的葉子幾點上面
        }else{
            root.right = insertNode(root.right,node);
        }
        return root;//最後返回的root值爲根節點,每次遞歸後就要返回當前的root值,以備上一層使用,最後返回整個樹的根節點
    }
    
}
相關文章
相關標籤/搜索