I think he was asked to write a function that will return true if the tree is balanced, not to actually balances it, which in my opinion is an exponentially easier question.