Leetcode Add To Array Python