Data Science Interview Leetcode Python