IOS封装自定义布局的方法

5）、设置collectionView内容的尺寸

- (CGSize)collectionViewContentSize
{
  //获取最高的那一列的高度
  CGFloat maxColumnHeight = [self.columnHeights[0] doubleValue];
  for (NSInteger i = 1; i<self.columnCount; i++)
  {
    CGFloat columnHeight = [self.columnHeights[i] doubleValue];
    if (maxColumnHeight < columnHeight)
    {
      maxColumnHeight = columnHeight;
    }
  }
  //返回collectionView的contentSize，高度为最高的高度加上一个行间距
  return CGSizeMake(0, maxColumnHeight + self.rowMargin);
}