判断二叉树是否是二叉排序树 思路与代码

问题描述

给定一棵二叉树的根节点 root，请判断其是否是二叉树排序树？

二叉排序树的基本概念：

1. 如果二叉树根节点的左子树不为空 ，那么左子树的所有节点数值均不大于根节点数值；
2. 如果二叉树根节点的右子树不为空 ，那么右子树的所有节点数值均不小于根节点数值；
3. 根节点的左子树、右子树也均为二叉排序树。

问题分析

此处需要用到树的遍历，二叉树常用的遍历操作为前序遍历、中序遍历、后续遍历、层序遍历等遍历方式，此处我们用到二叉树的中序遍历方式，根据中序遍历二叉树的逻辑，中序遍历的结果刚好是二叉树所有节点数值从小到大排序的结果，因此只需要判断遍历的结果是否满足从小到大的顺序关系即可判断该二叉树是否是二叉排序树。

问题解法

代码如下：

#  二叉树的基本结构
class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

def is_sort_binary_tree(root):
    while root or s: 
        while root:
            s.append(root)
            root = root.left
        root = s.pop()
        res.append(root.val)
        if len(res) >= 2 and res[-2] > res[-1]:
            return False
        root = root.right
    return True

用下面的代码测试：

if __name__ == "__main__":
    """
                4
            2       6
        1       3
        """
    root = TreeNode(4)
    root.left = layer2_left = TreeNode(2)
    root.right = layer2_right = TreeNode(6)
    layer2_left.left = TreeNode(1)
    layer2_left.right = TreeNode(3)

    print(is_sort_binary_tree(root))

结果返回 True，也可以用其他测试实例进行测试，结果正确。