如何在UITableView中显示2个自定义单元格

let tableView = UITableView()
tableView.delegate = self
tableView.dataSource = self

func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
    // 返回表格的行数
    return 2
}

func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
    // 返回自定义的单元格
    if indexPath.row == 0 {
        // 第一个自定义单元格
        let cell = tableView.dequeueReusableCell(withIdentifier: "CustomCell1", for: indexPath) as! CustomCell1
        // 配置第一个自定义单元格的内容
        return cell
    } else {
        // 第二个自定义单元格
        let cell = tableView.dequeueReusableCell(withIdentifier: "CustomCell2", for: indexPath) as! CustomCell2
        // 配置第二个自定义单元格的内容
        return cell
    }
}

tableView.register(CustomCell1.self, forCellReuseIdentifier: "CustomCell1")
tableView.register(CustomCell2.self, forCellReuseIdentifier: "CustomCell2")

func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
    // 返回单元格的高度
    return 50
}