700. Search in a Binary Search Tree

Stanislav Gumeniuk включен в leetcode leetcode-problem

2020.06.15 147 слов Одна минута

Содержание

Задача

Дан корневой узел бинарного дерева поиска (BST) и значение. Необходимо найти это значение в этом дереве, и если оно представлено - вернуть узел с этим значением.

Например, дано дерево:

Найти значение: 2 необходимо вернуть поддерево:

Решение

BST

Используем имплементацию .

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23


/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func searchBST(root *TreeNode, val int) *TreeNode {
    if root == nil {
        return nil
    }
    
    if root.Val == val {
        return root
    }
    
    if root.Val > val {
        return searchBST(root.Left, val)
    }
    
    return searchBST(root.Right, val)
}