Pregunta de entrevista de Google

tree related problem, leetcode Easy to medium