一、创建WaterFallLayout，继承于UICollectionViewFlowLayout

二、创建属性

var itemCount:Int  

var attributeArray:Array<UICollectionViewLayoutAttributes>?

三、添加自定义构造方法

required init?(coder aDecoder:NSCoder){

itemCount = 0

super.init(coder:aDecoder)

}

//自定义一个构造方法

init(itemCount:Int){

self.itemCount = itemCount

super.init()

}

四、

五、

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

return attributeArray

}