리트코드 700. Search in a Binary Search Tree

29 Nov 2022 in Algorithm

BST 에서 주어지는 숫자를 찾아라!

Tree 문제.. 🎋

문제

700. Search in a Binary Search Tree 문제 링크

BST 에서 주어지는 숫자를 찾아라!

# Example 1:
Input: root = [4,2,7,1,3], val = 2
Output: [2,1,3]

단순하게 노드에서 val에 해당하는 값이 있다면 해당 노드를 반환하면 됩니다.

✅ 최종 Solution

간단하게 재귀를 이용해서 풀었으며, 왼쪽 오른쪽 각각 재귀로 반복하게하여 해당 값이 있다면 노드를 반환하도록 구성했습니다.

def searchBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:
  if not root: return None
  if root.val == val:
    return root
  return self.searchBST(root.left, val) or self.searchBST(root.right, val)