Pergunta de entrevista da empresa Apple

Tree related leetcode question (medium)