二叉搜索树中的插入操作

发表于
本文字数： 1.3k 阅读时长 ≈ 1 分钟

LeetCode每日一题，701. Insert Into a Binary Search Tree

先看题目描述

大意就是给定一棵二叉搜索树和一个节点，让我们把节点插入二叉搜索树中，并将二叉树返回

算法思路

模拟

这题利用二叉搜索树的性质就可以，当将 val 插入到以 root 为根的子树上时，根据 val 与 root.val 的大小关系，就可以确定要将 val 插入到哪个子树中

  • 如果该子树不为空，则问题转化成了将 val 插入到对应子树上
  • 否则，在此处新建一个以 val 为值的节点，并链接到其父节点 root 上

算法源码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) return new TreeNode(val);
        dfs(root, val);
        return root;
    }

    private void dfs(TreeNode root, int val) {
        if (val < root.val) {
            if (root.left == null) {
                root.left = new TreeNode(val);
            } else {
                dfs(root.left, val);
            }
        } else {
            if (root.right == null) {
                root.right = new TreeNode(val);
            } else {
                dfs(root.right, val);
            }
        }
    }
}