LeetCode700. 二叉搜索树中的搜索-迭代与递归-java与python代码（超简单的写法！！！！）

https://leetcode-cn.com/problems/search-in-a-binary-search-tree/

题意

给定二叉搜索树（BST）的根节点和一个值。 你需要在BST中找到节点值等于给定值的节点。 返回以该节点为根的子树。 如果节点不存在，则返回NULL。

题解

简单题直接做就完事了。

递归方法

java代码

//递归解法
class Solution {
    public TreeNode searchBST(TreeNode root, int val) {
        if(null==root||root.val==val)return root;
        else if(root.val>val)return  searchBST(root.left,val);
        else return searchBST(root.right,val);
    }
}

python代码

class TreeNode:
     def __init__(self, val=0, left=None, right=None):
         self.val = val
         self.left = left
         self.right = right

class Solution:
    def searchBST(self, root: TreeNode, val: int) -> TreeNode:
        if not root or root.val==val: return root
        elif root.val>val: return self.searchBST(root.left,val)
        else: return self.searchBST(root.right,val)

迭代方法

java代码

 迭代解法
 class Solution {
     public TreeNode searchBST(TreeNode root, int val) {
         while(true) {
             if (root==null||val == root.val) return root;
             root=root.val>val?root.left:root.right;
         }
     }
 }

python代码

class TreeNode:
     def __init__(self, val=0, left=None, right=None):
         self.val = val
         self.left = left
         self.right = right

class Solution:
    def searchBST(self, root: TreeNode, val: int) -> TreeNode:
        while 1:
            if None==root or val==root.val:return root
            elif val<root.val:root=root.left
            else:root=root.right