LeetCode.700.二叉搜索树——java

二叉搜索树：

题目描述：

给定二叉搜索树（BST）的根节点 root 和一个整数值 val。

你需要在 BST 中找到节点值等于 val 的节点。 返回以该节点为根的子树。 如果节点不存在，则返回 null 。

示例 1:

输入：root = [4,2,7,1,3], val = 2 输出：[2,1,3] Example 2:

输入：root = [4,2,7,1,3], val = 5 输出：[]

解法一：迭代法——广度优先搜索

/**
 * 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 searchBST(TreeNode root, int val) {
             while(root!=null)
             {
                 if(root.val>val)
                 {
                     root=root.left;
                 }else if(root.val<val)
                 {
                     root=root.right;
                 }else{
                     return root;
                 }
             }
             return null;
    }
}

解法二：递归法——深度优先搜索：
递归返回值的问题：

/**
 * 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 searchBST(TreeNode root, int val) {
          if(root==null||root.val==val)
          {
              return root;
          }
          if(root.val>val)
          {
              return searchBST(root.left,val);
          }else if(root.val<val)
          {
              return searchBST(root.right,val);
          }
          return null;
    }
}