Gjenero një funksion për të kontrolluar nëse një pemë binare është e ekuilibruar apo jo.
def isBalanced ((root): nëse rrënja është Asnjë: ktheni True lh = height ((root.left) rh = height ((root.right) nëse (abs ((lh - rh) <= 1) dhe isBalanced ((root.left) është True dhe isBalanced ((root.right) është True: ktheni True ktheni False def height ((node): nëse nyja është None: ktheni 0 ktheni 1 + ((height ((node.left), height maxnode.right))