Leetcode 410 Split Array Python Initialize 2d