Question d’entretien chez Amazon

medium leetcode question, with array