Swift4返回主控制器后取消CollectionView选中复选框的方法

这个问题其实很常见，核心是要做好选中状态的数据源管理和返回时的状态重置，结合你用的CheckMarkView库，我给你几个具体的解决方案：

第一步：先维护一个选中状态的数据源 #

首先你需要在主视图控制器里，用一个数据源来记录每个CollectionView cell的复选框状态，而不是直接依赖cell里的UI组件（因为cell是复用的，直接操作UI会导致状态混乱）。比如：

// 主VC里定义：记录每个indexPath对应的选中状态
var selectedCheckStates: [IndexPath: Bool] = [:]

然后在collectionView(_:cellForItemAt:)方法里，根据这个数据源来设置CheckMarkView的状态，同时绑定切换事件：

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
    let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "你的CellID", for: indexPath) as! 你的自定义Cell类型
    
    // 根据数据源设置复选框状态，默认未选中
    let isChecked = selectedCheckStates[indexPath] ?? false
    cell.checkMarkViews.isChecked = isChecked
    
    // 绑定复选框切换事件，更新数据源
    cell.checkMarkViews.onToggle = { [weak self] isChecked in
        self?.selectedCheckStates[indexPath] = isChecked
    }
    
    return cell
}

第二步：返回主视图时重置状态 #

接下来分两种跳转场景处理：

场景1：通过导航栏push/pop跳转 #

当你从子视图控制器返回时，主VC的viewWillAppear方法会被触发，在这里重置状态并刷新CollectionView：

override func viewWillAppear(_ animated: Bool) {
    super.viewWillAppear(animated)
    
    // 清空选中状态数据源
    selectedCheckStates.removeAll()
    // 刷新CollectionView，让所有cell重新加载为未选中状态
    collectionView.reloadData()
}

场景2：通过模态present/dismiss跳转 #

如果是模态弹出子VC，除了用上面的viewWillAppear方法，还可以通过子VC的回调来更精准地处理：

在子视图控制器里定义一个回调闭包：

// 子VC里
var onDismissCallback: (() -> Void)?

@IBAction func backButtonTapped(_ sender: UIButton) {
    dismiss(animated: true) {
        // 触发回调通知主VC
        self.onDismissCallback?()
    }
}

然后在主VC跳转子VC时绑定回调：

// 主VC里跳转子VC的代码
let detailVC = storyboard?.instantiateViewController(withIdentifier: "DetailVC") as! 你的子VC类型
detailVC.onDismissCallback = { [weak self] in
    self?.selectedCheckStates.removeAll()
    self?.collectionView.reloadData()
}
present(detailVC, animated: true)
// 如果是push的话：navigationController?.pushViewController(detailVC, animated: true)

额外注意事项 #

• 一定要通过数据源管理状态，不要直接操作cell的CheckMarkView，否则cell复用时会出现状态错乱的问题。
• 调用collectionView.reloadData()会让所有cell重新加载，确保每个复选框都回到未选中状态。

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