如何更高效估算UICollectionView中Cell的高度？

我在UICollectionView中实现了无限滚动，发现当前估算Cell高度的方式成为性能瓶颈，滚动时数据量越大延迟越明显。请问有没有更优的Cell高度估算方法？

我的每个Cell内都包含一个UILabel，会为其分配不同长度的NSMutableAttributedString，代码如下：

let paragraphStyle = NSMutableParagraphStyle()
paragraphStyle.alignment = .justified
paragraphStyle.lineSpacing = 5.0
let attributedText = NSMutableAttributedString(string: " \(post.caption)", attributes: [NSAttributedStringKey.font: UIFont.systemFont(ofSize: 15), .paragraphStyle: paragraphStyle, .baselineOffset: NSNumber(value: 0)])
attributedText.append(NSAttributedString(string: "\n\n", attributes: [NSAttributedStringKey.font: UIFont.systemFont(ofSize: 4)]))
let timeAgoDisplay = post.creationDate.timeAgoDisplay()
attributedText.append(NSAttributedString(string: timeAgoDisplay, attributes: [NSAttributedStringKey.font: UIFont.systemFont(ofSize: 14), NSAttributedStringKey.foregroundColor: UIColor.storiesLightGray()]))
captionLabel.attributedText = attributedText

当前我使用sizeForItemAt方法估算高度，但调用dummyCell.layoutIfNeeded()后，当数据超过200项时应用变得非常卡顿，代码如下：

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
    var height: CGFloat = 180
    let frame = CGRect(x: 0, y: 0, width: view.frame.width, height: height)
    let dummyCell = HomePostCell(frame: frame)
    dummyCell.post = presenter.posts[indexPath.item]
    dummyCell.layoutIfNeeded()
    let targetSize = CGSize(width: view.frame.width, height: 5000)
    let estimatedSize = dummyCell.systemLayoutSizeFitting(targetSize)
    let newHeight = max(height, estimatedSize.height)
    return CGSize(width: view.frame.width, height: newHeight)
}

老哥，我太懂这种卡顿的痛苦了——每次滚动都卡一下，数据多了简直没法用。你的问题核心在于每次计算Cell高度都创建新的dummyCell，还触发完整的视图布局计算，这在数据量大时完全是性能灾难。给你几个亲测有效的优化方案：

1. 预计算+缓存高度（最推荐） #

直接把高度计算从滚动阶段提前到数据加载阶段，之后sizeForItemAt只需要取缓存值，完全避免重复计算。

具体步骤： #

• 给你的Post模型加一个cachedHeight属性（或者单独维护一个字典[Int: CGFloat]存储索引和高度的对应关系）
• 编写一个单独计算文本高度的方法，完全复刻Cell里Label的样式和约束条件：

// 单独计算高度的方法，和Cell内文本样式完全对齐
func calculateCellHeight(for post: Post, cellWidth: CGFloat) -> CGFloat {
    // 1. 构建和Cell里完全一样的AttributedString
    let paragraphStyle = NSMutableParagraphStyle()
    paragraphStyle.alignment = .justified
    paragraphStyle.lineSpacing = 5.0
    
    let attributedText = NSMutableAttributedString(string: " \(post.caption)", attributes: [
        NSAttributedStringKey.font: UIFont.systemFont(ofSize: 15),
        .paragraphStyle: paragraphStyle,
        .baselineOffset: NSNumber(value: 0)
    ])
    attributedText.append(NSAttributedString(string: "\n\n", attributes: [
        NSAttributedStringKey.font: UIFont.systemFont(ofSize: 4)
    ]))
    let timeAgoDisplay = post.creationDate.timeAgoDisplay()
    attributedText.append(NSAttributedString(string: timeAgoDisplay, attributes: [
        NSAttributedStringKey.font: UIFont.systemFont(ofSize: 14),
        NSAttributedStringKey.foregroundColor: UIColor.storiesLightGray()
    ]))
    
    // 2. 直接计算文本高度，不需要创建Cell
    let textSize = attributedText.boundingRect(
        with: CGSize(width: cellWidth - 你的Label左右边距总和, height: CGFloat.greatestFiniteMagnitude),
        options: [.usesLineFragmentOrigin, .usesFontLeading],
        context: nil
    ).size
    
    // 3. 加上Cell内其他固定高度元素的总和（比如图片、上下边距、按钮等）
    // 假设你Cell里固定元素总高度是120，就返回120 + textSize.height
    return 120 + ceil(textSize.height) // 用ceil避免小数高度导致的布局问题
}

• 在数据加载完成后批量计算并缓存高度：

// 比如在获取到posts数据后调用
let cellWidth = view.frame.width
for (index, post) in presenter.posts.enumerated() {
    post.cachedHeight = calculateCellHeight(for: post, cellWidth: cellWidth)
}

• 最后在sizeForItemAt里直接取缓存：

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
    let post = presenter.posts[indexPath.item]
    return CGSize(width: view.frame.width, height: post.cachedHeight)
}

2. 复用单个DummyCell（快速改进） #

如果暂时不想改预计算逻辑，至少不要每次都新建DummyCell——创建视图对象的开销非常大。你可以在ViewController里维护一个全局的DummyCell实例，重复使用：

// 在ViewController里声明全局变量
private lazy var dummyCell: HomePostCell = {
    let frame = CGRect(x: 0, y: 0, width: view.frame.width, height: 0)
    let cell = HomePostCell(frame: frame)
    return cell
}()

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
    let post = presenter.posts[indexPath.item]
    dummyCell.post = post
    dummyCell.layoutIfNeeded()
    
    let targetSize = CGSize(width: view.frame.width, height: CGFloat.greatestFiniteMagnitude)
    // 明确设置优先级，让计算更准确
    let estimatedSize = dummyCell.systemLayoutSizeFitting(
        targetSize,
        withHorizontalFittingPriority: .required,
        verticalFittingPriority: .fittingSizeLevel
    )
    
    return CGSize(width: view.frame.width, height: estimatedSize.height)
}

这样每次计算都复用同一个Cell，避免了频繁创建销毁视图的开销，能快速缓解卡顿。

3. 改用UICollectionViewCompositionalLayout（iOS 13+） #

如果你的App支持iOS 13及以上，强烈推荐用CompositionalLayout，它对动态高度的原生支持更高效，还能自动缓存高度，不需要自己实现sizeForItemAt：

// 创建布局的方法
func createCollectionViewLayout() -> UICollectionViewLayout {
    // 用estimated高度让系统自动计算
    let itemSize = NSCollectionLayoutSize(
        widthDimension: .fractionalWidth(1.0),
        heightDimension: .estimated(180) // 给一个初始估算值
    )
    let item = NSCollectionLayoutItem(layoutSize: itemSize)
    
    let groupSize = NSCollectionLayoutSize(
        widthDimension: .fractionalWidth(1.0),
        heightDimension: .estimated(180)
    )
    let group = NSCollectionLayoutGroup.horizontal(layoutSize: groupSize, subitems: [item])
    
    let section = NSCollectionLayoutSection(group: group)
    // 可以设置section的边距等样式
    
    return UICollectionViewCompositionalLayout(section: section)
}

初始化CollectionView时使用这个布局，同时确保你的Cell内约束完整（Label的top、leading、trailing、bottom都有明确约束，让系统能自动推导高度）。这种方式下系统会智能管理高度缓存，大幅减少重复计算。

4. 额外优化细节 #

• 无限滚动时注意缓存清理：如果加载了上千条数据，缓存的高度会占用不少内存，你可以用LRU缓存（比如NSCache）来管理，或者当用户滚动到较新的位置时，清理掉前面已不可见的post的缓存高度。
• 避免在sizeForItemAt里做任何耗时操作：这个方法会被频繁调用，任何额外的计算都会放大性能问题。

内容的提问来源于stack exchange，提问作者Kirill