UICollectionView remove quebras de seção com UICollectionViewFlowLayout

Eu tenho um conjunto de dados dividido em várias seções, no entanto, gostaria de exibi-lo em um collectionView sem interrupções entre as seções. Aqui está uma ilustração do que eu quero alcançar:

Instead of:
0-0 0-1 0-2
0-3
1-0 1-1
2-0
3-0

I want:
0-0 0-1 0-2
0-3 1-0 1-1
2-0 3-0

Sei que a solução provavelmente está em uma subclasse UICollectionViewLayout personalizada, mas não tenho certeza de como conseguir algo assim.

obrigado

questionAnswers(1)

yourAnswerToTheQuestion