将复选标记保存到tableView Swift时出现问题

// 定义一个数组来保存选中的行的索引
var selectedRows: [Int] = []

// 在didSelectRowAt方法中更新选中状态并刷新tableView
func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
    let selectedRow = indexPath.row

    // 如果选中的行已存在于数组中，则从数组中移除
    if let index = selectedRows.firstIndex(of: selectedRow) {
        selectedRows.remove(at: index)
    } else {
        // 否则将选中的行添加到数组中
        selectedRows.append(selectedRow)
    }

    // 刷新tableView以更新视图
    tableView.reloadData()
}

// 在cellForRowAt方法中根据选中状态设置复选标记
func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    let cell = tableView.dequeueReusableCell(withIdentifier: "Cell", for: indexPath)
    
    // 检查当前行是否在选中的行数组中
    if selectedRows.contains(indexPath.row) {
        cell.accessoryType = .checkmark
    } else {
        cell.accessoryType = .none
    }
    
    return cell
}