Easy Array Questions Leetcode Python