Question d’entretien chez NVIDIA

Practical Python function question, not really leetcode.