Leetcode Basic Array Questions In Python