如何实现UICollectionView单元格自适应适配iPhone屏幕布局？

你的现有代码因为依赖简单的除法逻辑，没法精准匹配需求里的五种布局场景，所以超过6个单元格就会失效。我这里给你两个靠谱的实现方案，既能覆盖所有要求的场景，还能支持任意数量的单元格：

方案一：优化sizeForItemAt逻辑（推荐，简洁高效） #

这个方案直接针对你的布局规则做精准判断，逻辑清晰易懂，还能自动适配任意数量的单元格。

首先，我们需要先计算collectionView的可用宽高（用安全区域替代原来的statusBarFrame，能更好适配刘海屏、横屏等场景），然后写出完整的sizeForItemAt实现：

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
    let cellCount = self.contactsArray.count
    let availableWidth = collectionView.bounds.width
    // 计算扣除安全区域后的可用高度
    let availableHeight = collectionView.bounds.height 
                          - collectionView.safeAreaInsets.top 
                          - collectionView.safeAreaInsets.bottom
    
    switch cellCount {
    case 1:
        // 单个单元格填满整个可用区域
        return CGSize(width: availableWidth, height: availableHeight)
    case 2:
        // 两个单元格上下平分屏幕
        return CGSize(width: availableWidth, height: availableHeight / 2)
    case 3, 4:
        // 3或4个单元格：2x2的网格，3个时第4个位置留空
        return CGSize(width: availableWidth / 2, height: availableHeight / 2)
    default:
        // 5个及以上：每行固定2个单元格，行数随数量自动计算
        let totalRows = ceil(CGFloat(cellCount) / 2)
        let cellHeight = availableHeight / totalRows
        return CGSize(width: availableWidth / 2, height: cellHeight)
    }
}

配套的FlowLayout设置 #

别忘了给你的FlowLayout设置零间距，确保单元格完全贴合屏幕：

override func viewDidLoad() {
    super.viewDidLoad()
    if let flowLayout = collectionView.collectionViewLayout as? UICollectionViewFlowLayout {
        flowLayout.minimumInteritemSpacing = 0
        flowLayout.minimumLineSpacing = 0
        flowLayout.sectionInset = .zero
    }
}

这个方案不仅完美覆盖了你需求里的所有情况，超过6个单元格时，会自动按每行2个的规则扩展，每个单元格的高度会根据总行数均匀分配，完全适配屏幕边界。

方案二：自定义UICollectionViewFlowLayout（适合复杂扩展） #

如果未来你需要调整布局细节（比如给空位添加特殊样式、自定义单元格间距等），可以用自定义FlowLayout直接控制每个单元格的位置。这里给你一个适配需求的示例：

class CustomGridLayout: UICollectionViewFlowLayout {
    override func layoutAttributesForElements(in rect: CGRect) -> [UICollectionViewLayoutAttributes]? {
        guard let baseAttributes = super.layoutAttributesForElements(in: rect),
              let collectionView = collectionView else {
            return nil
        }
        
        let cellCount = collectionView.numberOfItems(inSection: 0)
        let availableWidth = collectionView.bounds.width
        let safeTop = collectionView.safeAreaInsets.top
        let availableHeight = collectionView.bounds.height - safeTop - collectionView.safeAreaInsets.bottom
        
        let adjustedAttributes = baseAttributes.map { attr -> UICollectionViewLayoutAttributes in
            let index = attr.indexPath.item
            var frame = attr.frame
            
            switch cellCount {
            case 1:
                frame = CGRect(x: 0, y: safeTop, width: availableWidth, height: availableHeight)
            case 2:
                let cellHeight = availableHeight / 2
                frame = CGRect(x: 0, y: safeTop + CGFloat(index) * cellHeight, width: availableWidth, height: cellHeight)
            case 3, 4:
                let cellSize = CGSize(width: availableWidth/2, height: availableHeight/2)
                let row = index / 2
                let col = index % 2
                frame = CGRect(x: CGFloat(col)*cellSize.width, y: safeTop + CGFloat(row)*cellSize.height, size: cellSize)
            default:
                let cellWidth = availableWidth / 2
                let totalRows = ceil(CGFloat(cellCount)/2)
                let cellHeight = availableHeight / totalRows
                let row = index / 2
                let col = index % 2
                frame = CGRect(x: CGFloat(col)*cellWidth, y: safeTop + CGFloat(row)*cellHeight, width: cellWidth, height: cellHeight)
            }
            
            attr.frame = frame
            return attr
        }
        
        return adjustedAttributes
    }
    
    // 屏幕尺寸变化时重新计算布局
    override func shouldInvalidateLayout(forBoundsChange newBounds: CGRect) -> Bool {
        return true
    }
}

使用的时候，直接给collectionView设置这个自定义Layout即可：

collectionView.collectionViewLayout = CustomGridLayout()

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