Question d’entretien chez TikTok

Task from the leetcode, average