Leetcode Split Array Python Definition