Array Leetcode Problems Python