如何根据UICollectionView的单元格数量调整视图尺寸？

嘿，这个需求我之前做项目的时候刚好碰到过，完全理解你不想固定高度导致内容显示不全或者留白过多的痛点！直接给你几个靠谱的实现方案，顺便说说为什么viewDidLoad()里没法直接搞定~

核心问题：为什么viewDidLoad()不行？ #

简单说，viewDidLoad()的时候，你的UICollectionView还没完成布局，数据源大概率也还没加载完成，此时它的contentSize还是默认值，根本没法准确拿到实际需要的高度。所以得等数据准备好、布局完成后再调整高度。

方案一：高度约束+动态更新（最常用） #

这个方法上手快，适合大多数场景：

1. 先给CollectionView加高度约束
   在Storyboard或者代码里，给UICollectionView添加一个高度约束，并且把它关联成IBOutlet，比如叫collectionViewHeightConstraint。

2. 数据加载完成后更新约束
   当你的数据源（比如那20个或5个菜单项）准备好之后，刷新CollectionView，然后强制触发布局，最后把高度约束的constant设为CollectionView的contentSize.height。

示例代码：

// 先声明变量
@IBOutlet weak var collectionView: UICollectionView!
@IBOutlet weak var collectionViewHeightConstraint: NSLayoutConstraint!
private var menuItems: [MenuItem] = []

// 加载数据的方法
func loadMenuItems() {
    // 模拟获取数据，比如从接口拿到菜单列表
    menuItems = fetchMenuItemsFromAPI()
    // 刷新CollectionView
    collectionView.reloadData()
    // 强制触发布局，确保contentSize计算准确
    collectionView.layoutIfNeeded()
    // 更新高度约束
    collectionViewHeightConstraint.constant = collectionView.contentSize.height
    // 可选：加个小动画让高度变化更自然
    UIView.animate(withDuration: 0.3) {
        self.view.layoutIfNeeded()
    }
}

3. 如果是动态高度的Cell
   如果你的Cell高度不是固定的，记得给FlowLayout设置自动计算尺寸：

   override func viewDidLoad() {
       super.viewDidLoad()
       if let flowLayout = collectionView.collectionViewLayout as? UICollectionViewFlowLayout {
           // 让Cell自己根据内容计算高度
           flowLayout.estimatedItemSize = UICollectionViewFlowLayout.automaticSize
       }
   }
   

方案二：自定义Self-Sizing CollectionView（更优雅） #

如果不想手动维护高度约束，可以自定义一个UICollectionView子类，让它自己根据内容调整高度：

class SelfSizingCollectionView: UICollectionView {
    override var intrinsicContentSize: CGSize {
        // 让固有尺寸等于内容尺寸
        return contentSize
    }
    
    override func layoutSubviews() {
        super.layoutSubviews()
        // 如果当前尺寸和固有尺寸不一样，就刷新固有尺寸
        if !bounds.size.equalTo(intrinsicContentSize) {
            invalidateIntrinsicContentSize()
        }
    }
}

然后在Storyboard里把你的CollectionView的类改成这个自定义类，接着给它添加上下左右的约束（比如和父视图的leading、trailing、top对齐，底部不需要约束），它就会自动根据单元格数量调整高度了！

注意事项 #

• 确保CollectionView的宽度是固定的（比如和父视图同宽），这样contentSize.height的计算才会准确。
• 如果用方案一，记得不要在viewDidLayoutSubviews里直接更新约束（会重复调用），最好加个判断，只有当数据源变化时才更新。

内容的提问来源于stack exchange，提问作者Ben B.