Pergunta de entrevista da empresa Aurora Innovation

Leetcode medium question which is not real world but an algorithm/DS type. Interviewer was nice but kept pushing for quick coding and optimal solution in first try.