Leetcode Medium Level Array In Python Declaration