Update 0700.二叉搜索树中的搜索.md

更改代码符合PEP 8 规范.
This commit is contained in:
Kelvin
2021-08-01 10:15:22 -04:00
parent 7b1c234d1f
commit a8109b4014

View File

@ -212,17 +212,20 @@ Python
递归法:
```python
# Definition for a binary tree node.
# 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 //为空或者已经找到都是直接返回root所以合并了
if root.val > val: return self.searchBST(root.left,val) //注意一定要加return
else: return self.searchBST(root.right,val)
# 为什么要有返回值:
# 因为搜索到目标节点就要立即return
# 这样才是找到节点就返回搜索某一条边如果不加return就是遍历整棵树了。
if not root or root.val == val:
return root
if root.val > val:
return self.searchBST(root.left, val)
if root.val < val:
return self.searchBST(root.right, val)
```