Tengo un conjunto de datos que se divide en varias secciones, sin embargo, me gustaría mostrar esto en una colecciónView sin interrupciones entre las secciones. Aquí hay una ilustración de lo que quiero lograr:

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

Me doy cuenta de que la solución probablemente reside en una subclase personalizada UICollectionViewLayout, pero no estoy seguro de cómo lograr algo como esto.

Gracias