封装自定义瀑布流，实现原理

继承UICollectionViewLayout
@interface WaterPullLayout : UICollectionViewLayout

需要两个数组来保存每个item的属性和每列的总高度

@interface WaterPullLayout ()
{
    /**
     *  保存每个item的属性
     */
    NSMutableArray *_itemAttributes;

    /**
     *  保存每列的高度
     */
    NSMutableArray *_columnHeights;
}

.m实现必须重写四个方法：

-//对item布局，reloadData会触发
- (void)prepareLayout
- //返回指定区域的item的属性对象数组
- (nullable NSArray<__kindof UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect
-//返回指定indexPath的item的属性对象
- (nullable UICollectionViewLayoutAttributes *)layoutAttributesForItemAtIndexPath:(NSIndexPath *)indexPath
-//collectionView的contentSize
- (CGSize)collectionViewContentSize

-#import 《UIKit/UIKit.h>
@protocol WaterPullLayoutDelegate

@required
- (CGSize)collectionView:(UICollectionView )collectionView layout:(UICollectionViewLayout )collectionViewLayout sizeForItemAtIndexPath:(NSIndexPath *)indexPath;

@end

@interface WaterPullLayout : UICollectionViewLayout

/**
* 列数
*/
@property (nonatomic) NSInteger columnCount;

/**
* 距离上下左右的距离
*/
@property (nonatomic) UIEdgeInsets sectionInsets;

/**
* 两个cell的水平方向的空隙
*/
@property (nonatomic) CGFloat horizontalSpace;

/**
* 两个cell的垂直方向的空隙
*/
@property (nonatomic) CGFloat verticalSpace;

@end

#import "WaterPullLayout.h"

@interface WaterPullLayout ()
{
    /**
     *  保存每个cell的属性
     */
    NSMutableArray *_itemAttributes;

    /**
     *  保存每列的高度
     */
    NSMutableArray *_columnHeights;
}

@property (nonatomic, weak) id<WaterPullLayoutDelegate> delegate;

@end

@implementation WaterPullLayout

- (instancetype)init
{
    self = [super init];
    if (self)
    {
        //默认2列
        _columnCount = 2;
        _horizontalSpace = 10;
        _verticalSpace = 10;
        _sectionInsets = UIEdgeInsetsZero;
    }
    return self;
}

/**
 *  返回高度最大的列的索引
 */
- (NSInteger)maxHeightColumnIndex
{
    __block NSInteger index = 0;
    __block CGFloat maxValue = 0;

    [_columnHeights enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop)
     {
         //如果从数组里面取出的元素比现有最大值还大，这个元素就是最大值
         if ([obj floatValue] > maxValue)
         {
             //替换最大值
             maxValue = [obj floatValue];

             //保存最大值的索引
             index = idx;
         }
     }];

    return index;
}

/**
 *  @return 返回高度最小的列的索引
 */
- (NSInteger)minHeightColumnIndex
{
    __block NSInteger index = 0;
    __block CGFloat minValue = MAXFLOAT;

    [_columnHeights enumerateObjectsUsingBlock:^(id  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop)
    {
        //如果从数组里面取出的元素比现有最小值还小，这个元素就是最小值
        if ([obj floatValue] < minValue)
        {
            //替换最小值
            minValue = [obj floatValue];

            //保存最小值的索引
            index = idx;
        }
    }];

    return index;
}

//对item布局，reloadData会触发
- (void)prepareLayout
{
    [super prepareLayout];

    self.delegate = (id<WaterPullLayoutDelegate>)self.collectionView.delegate;

    //初始化数组
    _itemAttributes = [NSMutableArray array];
    _columnHeights = [NSMutableArray array];

    //初始化默认高度
    for (int i = 0; i < self.columnCount; i++)
    {
        //每列的高度初始化都是0
        _columnHeights[i] = @(0);
    }

    //item的宽度 =（collectionView的宽 - 左边距 - 右边距 - item间的水平间隙 ）/ 列数
    CGFloat itemWidth = (self.collectionView.frame.size.width - self.sectionInsets.left - self.sectionInsets.right - (self.columnCount - 1)*self.horizontalSpace)/self.columnCount;


    //计算每个item的位置坐标(只考虑一个组的情况)
    //[self.collectionView numberOfItemsInSection:0]返回指定组的item个数
    for (int i = 0; i < [self.collectionView numberOfItemsInSection:0]; i++)
    {
        //获取item大小（调用collectionView的代理方法）
        CGSize size = [self.delegate collectionView:self.collectionView layout:self sizeForItemAtIndexPath:[NSIndexPath indexPathForItem:i inSection:0]];

        //item的高度
        CGFloat itemHeight;
        //item的x坐标
        CGFloat xOffset;
        //item的y坐标
        CGFloat yOffset;

        //现在图片的高度
        /*
         原始H    X
         ---- =  ----- ===>  X = 原始H*现在W/原始W
         原始W    现在W
         */
        itemHeight = size.height * itemWidth / size.width;

        //获取列的高度的最小值索引
        NSInteger minIndex = [self minHeightColumnIndex];

        NSLog(@"%ld",minIndex);

        //item的x坐标 = 与左边间距 + 最小高度索引值*（item宽 + item间水平间隙）
        xOffset = self.sectionInsets.left + minIndex*(itemWidth+self.horizontalSpace);

        //item的y坐标 = 最小高度 + （item个数如果小于列数）？与顶部间距 ：item间的垂直间隙
        yOffset = [_columnHeights[minIndex] floatValue] + ((i < self.columnCount) ? self.sectionInsets.top : self.verticalSpace);

        //item的frame
        CGRect itemFrame = CGRectMake(xOffset, yOffset, itemWidth, itemHeight);

        //属性
        UICollectionViewLayoutAttributes *attribute = [UICollectionViewLayoutAttributes layoutAttributesForCellWithIndexPath:[NSIndexPath indexPathForItem:i inSection:0]];
        attribute.frame = itemFrame;

        //添加到数组中
        [_itemAttributes addObject:attribute];


        //修改当前列的高度
        _columnHeights[minIndex] = @(CGRectGetMaxY(itemFrame));
    }
}

/**
 *  返回指定区域的cell的属性对象数组
 *
 *  @param rect <#rect description#>
 *
 *  @return <#return value description#>
 */
- (nullable NSArray<__kindof UICollectionViewLayoutAttributes *> *)layoutAttributesForElementsInRect:(CGRect)rect
{
    return _itemAttributes;
}

/**
 *  返回指定indexPath的item的属性对象
 *
 *  @param indexPath <#indexPath description#>
 */
- (nullable UICollectionViewLayoutAttributes *)layoutAttributesForItemAtIndexPath:(NSIndexPath *)indexPath
{
    return _itemAttributes[indexPath.item];
}

/**
 *  collectionView的contentSize
 */
- (CGSize)collectionViewContentSize
{
    //取最大值的索引
    NSInteger maxIndex = [self maxHeightColumnIndex];

    //返回collectionView的contentSize
    return CGSizeMake(self.collectionView.frame.size.width, [_columnHeights[maxIndex] floatValue] + self.sectionInsets.bottom);
}

@end