https://rishi.tips/t/array-partition/