为UITableView中的多个单元格指定不同的行数

class ViewController: UIViewController, UITableViewDataSource, UITableViewDelegate {
    // 定义一个数组来存储不同单元格的数据
    let data = ["Cell 1", "Cell 2", "Cell 3"]
    
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        // 返回不同section中的行数
        if section == 0 {
            return 2
        } else if section == 1 {
            return 3
        } else {
            return 1
        }
    }
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "Cell", for: indexPath)
        
        // 根据indexPath获取当前单元格的位置信息
        if indexPath.section == 0 {
            cell.textLabel?.text = data[indexPath.row]
        } else if indexPath.section == 1 {
            cell.textLabel?.text = "Section 2, Row \(indexPath.row)"
        } else {
            cell.textLabel?.text = "Section 3"
        }
        
        return cell
    }
    
    // 其他UITableViewDelegate协议中的方法...
}