实现瀑布流的效果我们可以通过重写 UICollectionViewLayout里面

- (void)prepareLayout;

- (NSArray *)layoutAttributesForElementsInRect:(CGRect)rect;

- (UICollectionViewLayoutAttributes *)layoutAttributesForItemAtIndexPath:(NSIndexPath *)indexPath;

- (CGSize)collectionViewContentSize;

的这4个方法。。。

下面是4个方法的具体实现

/**

 * 初始化

 */

- (void)prepareLayout

{

    [super prepareLayout];

    self.contentHeight = 0;

    // 清除以前计算的所有高度

    [self.columnHeights removeAllObjects];

    for (NSInteger i = 0; i < self.columnCount; i++) {

        [self.columnHeights addObject:@(self.edgeInsets.top)];

    }

    // 清除之前所有的布局属性

    [self.attrsArray removeAllObjects];

    // 开始创建每一个cell对应的布局属性

    NSInteger count = [self.collectionView numberOfItemsInSection:0];

    for (NSInteger i = 0; i < count; i++) {

        // 创建位置

        NSIndexPath *indexPath = [NSIndexPath indexPathForItem:i inSection:0];

        // 获取indexPath位置cell对应的布局属性

        UICollectionViewLayoutAttributes *attrs = [self layoutAttributesForItemAtIndexPath:indexPath];

        [self.attrsArray addObject:attrs];

    }

}

/**

 * 决定cell的排布

 */

- (NSArray *)layoutAttributesForElementsInRect:(CGRect)rect

{

    return self.attrsArray;

}

/**

 * 返回indexPath位置cell对应的布局属性

 */

- (UICollectionViewLayoutAttributes *)layoutAttributesForItemAtIndexPath:(NSIndexPath *)indexPath

{

    // 创建布局属性

    UICollectionViewLayoutAttributes *attrs = [UICollectionViewLayoutAttributes layoutAttributesForCellWithIndexPath:indexPath];

    // collectionView的宽度

    CGFloat collectionViewW = self.collectionView.frame.size.width;

    // 设置布局属性的frame

    CGFloat w = (collectionViewW - self.edgeInsets.left - self.edgeInsets.right - (self.columnCount - 1) * self.columnMargin) / self.columnCount;

    CGFloat h = [self.delegate waterflowLayout:self heightForItemAtIndex:indexPath.item itemWidth:w];

    // 找出高度最短的那一列

    NSInteger destColumn = 0;

    CGFloat minColumnHeight = [self.columnHeights[0] doubleValue];

    for (NSInteger i = 1; i < self.columnCount; i++) {

        // 取得第i列的高度

        CGFloat columnHeight = [self.columnHeights[i] doubleValue];

        if (minColumnHeight > columnHeight) {

            minColumnHeight = columnHeight;

            destColumn = i;

        }

    }

    CGFloat x = self.edgeInsets.left + destColumn * (w + self.columnMargin);

    CGFloat y = minColumnHeight;

    if (y != self.edgeInsets.top) {

        y += self.rowMargin;

    }

    attrs.frame = CGRectMake(x, y, w, h);

    // 更新最短那列的高度

    self.columnHeights[destColumn] = @(CGRectGetMaxY(attrs.frame));

    // 记录内容的高度

    CGFloat columnHeight = [self.columnHeights[destColumn] doubleValue];

    if (self.contentHeight < columnHeight) {

        self.contentHeight = columnHeight;

    }

    return attrs;

}

- (CGSize)collectionViewContentSize

{

    return CGSizeMake(0, self.contentHeight + self.edgeInsets.bottom);

}

效果如下图：

UICollectionView瀑布流的实现