about summary refs log tree commit diff
path: root/users/wpcarro/scratch/facebook/bst-checker.py
blob: 7ef63a95315ec9e8fc917f4686ed20bfc2939303 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
from collections import deque

class Node(object):
    def __init__(self, value, left=None, right=None):
        self.value = value
        self.left = left
        self.right = right

    def is_bst(self):
        s = []
        s.append((float('-inf'), self, float('inf')))
        while s:
            lo, node, hi = s.pop()
            if lo <= node.value <= hi:
                node.left and s.append((lo, node.left, node.value))
                node.right and s.append((node.value, node.right, hi))
            else:
                return False
        return True


x = Node(
    50,
    Node(
        17,
        Node(
            12,
            Node(9),
            Node(14),
        ),
        Node(
            23,
            Node(19),
        ),
    ),
    Node(
        72,
        Node(
            54,
            None,
            Node(67)
        ),
        Node(76),
    ),
)


assert x.is_bst()
print("Success!")