Staggered Layout in UICollectionView (iOS 10 and above) - iOS Swift

Question

Hi, This is my dataSource for UICollectionView

var dataSource = ["1", "2", "3", "4", "5"