Pregunta de entrevista de Wish

leetcode 844 and a medium tree problem