from AVLNode import AVLNode

class ExtendedAVLNode$($AVLNode$)$:

def $\_\_$init$\_\_($self$,$ node$\_$key$)$:

super$().\_\_$init$\_\_($node$\_$key$)$

self.subtree$\_$key$\_$count $=1$

def get$\_$subtree$\_$key$\_$count$($self$)$:

return self.subtree$\_$key$\_$count

def update$\_$subtree$\_$key$\_$count$($self$)$:

self.subtree$\_$key$\_$count $=1$

if self.left:

self.subtree$\_$key$\_$count $+=$ self.left.subtree$\_$key$\_$count

if self.right:

self.subtree$\_$key$\_$count $+=$ self.right.subtree$\_$key$\_$count $\backslash$