如何在Swift3中使用枚举分配UITableView中的自定义单元格？

enum CellType {
    case normal
    case custom
}

func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    let cellType: CellType = indexPath.row % 2 == 0 ? .normal : .custom
    
    switch cellType {
    case .normal:
        let cell = tableView.dequeueReusableCell(withIdentifier: "NormalCell", for: indexPath) as! NormalTableViewCell
        // 配置普通单元格的内容
        return cell
    case .custom:
        let cell = tableView.dequeueReusableCell(withIdentifier: "CustomCell", for: indexPath) as! CustomTableViewCell
        // 配置自定义单元格的内容
        return cell
    }
}

func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
    let cellType: CellType = indexPath.row % 2 == 0 ? .normal : .custom
    
    switch cellType {
    case .normal:
        return 44 // 普通单元格的高度
    case .custom:
        return 100 // 自定义单元格的高度
    }
}