Return root node given shuffled nodes - SWE Interview (with Principal Software Engineer)