Search in a Binary Search Tree

,

Problem

Given the root node of a binary search tree (BST) and a value. You need to find the node in the BST that the node’s value equals the given value. Return the subtree rooted with that node. If such node doesn’t exist, you should return NULL.

Example

1
2
3
4
5
6
7
8
9
10
11

Input: 
        4
       / \
      2   7
     / \
    1   3
    search : 2
Output: 
      2     
     / \   
    1   3

My Answer

  • BST 이기 때문에 nxt.val과 val의 차이에 따라 nxt혹은 결과가 나온다.
  • 만약 nxt.val이 val보다 작다면, 정답은 nxt의 우측에 있다는 의미
  • 만약 nxt.val이 val보다 크다면, 정답은 nxt의 좌측에 있다는 의미
  • 만약 nxt.val이 val과 같다면, 정답
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

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode searchBST(TreeNode root, int val) {
        TreeNode res = null;
        TreeNode nxt = root;
        
        while ( nxt != null ) {
            if ( nxt.val == val ) {
                res = nxt;
                break;
            }    
            
            if ( nxt.val < val ) nxt = nxt.right;
            else nxt = nxt.left;
        }
        
        return res;
    }
}

참고