算法——dfs 判断是否为BST

95. 验证二叉查找树

中文English

给定一个二叉树,判断它是否是合法的二叉查找树(BST)

一棵BST定义为:

  • 节点的左子树中的值要严格小于该节点的值。
  • 节点的右子树中的值要严格大于该节点的值。
  • 左右子树也必须是二叉查找树。
  • 一个节点的树也是二叉查找树。

Example

样例 1:

输入:{-1}
输出:true
解释:
二叉树如下(仅有一个节点):
	-1
这是二叉查找树。

样例 2:

输入:{2,1,4,#,#,3,5}
输出:true
解释:
	二叉树如下:
	  2
	 / \
	1   4
	   / \
	  3   5
这是二叉查找树。

"""
Definition of TreeNode:
class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None
"""

class Solution:
    """
    @param root: The root of binary tree.
    @return: True if the binary tree is BST, or false
    """
    _is_bst = True
    _last_val = -float('inf')
    
    def isValidBST(self, root):
        # write your code here
        self.helper(root)
        return self._is_bst
        
    def helper(self, root):
        if not root:
            return
        self.helper(root.left)
        if root.val <= self._last_val:
            self._is_bst = False # return
        self._last_val = root.val
        self.helper(root.right)
    

  

需要在遍历中记住上次遍历节点,根据上次的节点和当前节点进行比较而得到result的算法模板:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class Solution():
   last_node = None
   result = None
    
   def run(self, root):
        self.dfs(root)
        return self.result
    
   def dfs(self):
       if last_node is None:
           last_node = root
       else:
           do_sth(last_node, root)
            
       dfs(root.left)
 
       dfs(root.right)
posted @ 2019-09-29 21:09  bonelee  阅读(722)  评论(0编辑  收藏  举报