700. 二叉搜索树中的搜索 简单

给定二叉搜索树(BST)的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在,则返回 NULL。

例如,

给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和值: 2

你应该返回如下子树:

      2     
     / \   
    1   3

在上述示例中,如果要找的值是 5,但因为没有节点值为 5,我们应该返回 NULL。

代码参考:

package main

import "fmt"

func main() {
    root := &TreeNode{Val: 4}
    root.Left = &TreeNode{Val: 2}
    root.Left.Left = &TreeNode{Val: 1}
    root.Left.Right = &TreeNode{Val: 3}
    root.Right = &TreeNode{Val: 7}

    //        4
    //       / \
    //      2   7
    //     / \
    //    1   3
    fmt.Println(searchBST(root, 2)) // 2
}

// what can I say
func searchBST(root *TreeNode, val int) *TreeNode {
    if root != nil {
        switch {
        case root.Val < val:
            return searchBST(root.Right, val)
        case root.Val > val:
            return searchBST(root.Left, val)
        default:
            return root
        }
    }
    return nil
}
最后编辑: kuteng  文档更新时间: 2021-06-05 10:16   作者:kuteng