https://remp3indir.net/leetcode-convert-1d-array-into-2d-array-python/