CollectionView单元格按选中顺序匹配颜色数组，需在isHighlighted前传值

我懂你遇到的这个麻烦——想让CollectionView的选中单元格按顺序对应颜色数组，但系统的选择回调总是赶在isHighlighted设置之后，遍历所有单元格又太冗余。这里有个更清爽的解决方案，不用依赖那些时机不对的回调，而是通过维护选中状态的有序集合来驱动单元格颜色：

核心思路 #

我们需要记录选中单元格的顺序，然后在单元格配置时，根据它在选中列表中的位置匹配颜色数组对应的颜色。这样不管是选中还是取消选中，只要更新这个有序集合并刷新对应单元格，就能自动同步颜色。

具体实现步骤 #

1. 定义状态变量和颜色数组 #

首先在你的ViewController里维护一个有序的选中索引数组，以及你的目标颜色数组：

let colorArray: [UIColor] = [.blue, .green, .red]
var selectedIndices: [IndexPath] = [] // 按选中顺序存储索引

2. 自定义CollectionViewCell #

给单元格加一个颜色属性，用来直接控制背景色，避免依赖isHighlighted的时机问题：

class CustomCollectionViewCell: UICollectionViewCell {
    // 自定义颜色属性，设置时自动更新背景
    var cellColor: UIColor? {
        didSet {
            backgroundColor = cellColor ?? .clear
        }
    }
    
    // 如果需要高亮反馈，可以重写isHighlighted，优先保留选中颜色
    override var isHighlighted: Bool {
        didSet {
            if isHighlighted && cellColor == nil {
                // 未选中时的临时高亮色
                backgroundColor = .lightGray
            } else {
                // 选中状态下保持对应颜色，未选中则恢复透明
                backgroundColor = cellColor ?? .clear
            }
        }
    }
}

3. 在数据源方法中配置单元格颜色 #

在cellForItemAt里，根据当前单元格的索引是否在选中列表中，找到它的位置并匹配颜色：

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
    let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "CustomCell", for: indexPath) as! CustomCollectionViewCell
    
    // 找到当前索引在选中列表中的位置
    if let position = selectedIndices.firstIndex(of: indexPath) {
        // 确保不超出颜色数组的范围
        cell.cellColor = position < colorArray.count ? colorArray[position] : .clear
    } else {
        cell.cellColor = .clear
    }
    
    return cell
}

4. 处理选中/取消选中的逻辑 #

在didSelectItemAt和didDeselectItemAt里更新选中索引数组，然后刷新需要变更的单元格：

func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
    guard !selectedIndices.contains(indexPath) else { return }
    
    selectedIndices.append(indexPath)
    // 刷新所有选中的单元格，确保颜色顺序正确
    collectionView.reloadItems(at: selectedIndices)
}

func collectionView(_ collectionView: UICollectionView, didDeselectItemAt indexPath: IndexPath) {
    guard let removeIndex = selectedIndices.firstIndex(of: indexPath) else { return }
    
    selectedIndices.remove(at: removeIndex)
    // 先刷新被取消的单元格，恢复透明
    collectionView.reloadItems(at: [indexPath])
    // 刷新剩下的选中单元格，因为它们的位置前移了，需要匹配新的颜色位置
    collectionView.reloadItems(at: selectedIndices)
}

为什么这个方案更好？ #

• 完全摆脱了回调时机的限制，逻辑全部基于数据源状态，更符合MVC的设计思路
• 不需要遍历所有可见单元格，只刷新需要更新的单元格，性能更优
• 选中顺序和颜色的对应关系清晰，后续扩展（比如颜色数组长度变化）也更容易调整

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