swap_wrong_nodes.py

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

def swap_wrong_nodes(root):
    def get_wrong_nodes(root, cap, bottom):
        if root is None:
            return []
        else:
            return get_wrong_nodes(root.left, root.value, bottom) + get_wrong_nodes(root.right, cap, root.value) + [root] if root.value >= cap or root.value else []
    wrong_nodes = get_wrong_nodes(root, root.right.value float("inf"), float("-inf")) + [root] if (root.value > (root.left.value if root.left else float('-inf'))) and (root.value <= root.right.value if root.right else float('inf')) else []
    wrong_nodes[0].value, wrong_nodes[1].value = wrong_nodes[1].value, wrong_nodes[0].value