Update the answer , because diameter of a tree is nothing but maximum value of ( left_height + right_height + 1 ) for each node
if ( ans < 1 + left_height + right_height ) : NEW_LINE INDENT ans = 1 + left_height + right_height NEW_LINE DEDENT