Leetcode刷题700. 二叉搜索树中的搜索

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

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

示例 1:

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

示例 2:

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/search-in-a-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
    public TreeNode searchBST(TreeNode root, int val) {
//		return searchBSTI(root, val);
		return searchBSTII(root, val);
	}

	//方法二：迭代
	//时间复杂度O(N)，空间复杂度O(1)
	private TreeNode searchBSTII(TreeNode root, int val) {
		if (root == null) {
			return null;
		}
		while (root != null) {
			if (root.val == val) {
				break;
			} else if (root.val < val) {
				root = root.right;
			} else {
				root = root.left;
			}
		}
		return root;
	}


	//方法一：利用BST中序遍历有序，递归
	//时间和空间复杂度O(N)
	private TreeNode searchBSTI(TreeNode root, int val) {
		if (root == null) {
			return null;
		}
		if (root.val < val) {
			return searchBSTI(root.right, val);
		} else if (root.val > val) {
			return searchBSTI(root.left, val);
		} else {
			return root;
		}
	}
}