一、創(chuàng)建WaterFallLayout，繼承于UICollectionViewFlowLayout

二、創(chuàng)建屬性

var itemCount:Int??

var attributeArray:Array<UICollectionViewLayoutAttributes>?

三、添加自定義構(gòu)造方法

required init?(coder aDecoder:NSCoder){

itemCount = 0

super.init(coder:aDecoder)

}

//自定義一個構(gòu)造方法

init(itemCount:Int){

self.itemCount = itemCount

super.init()

}

四、

五、

func layoutAttributesForElements(in rect : CGRect) -> [UICollectionViewLayoutAttributes]?{

return attributeArray

}