表单元格不能根据uiimageview高度自动调整大小

class CustomTableViewCell: UITableViewCell {
    var customImageView: UIImageView!

    override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
        super.init(style: style, reuseIdentifier: reuseIdentifier)
        
        customImageView = UIImageView()
        customImageView.contentMode = .scaleAspectFit
        contentView.addSubview(customImageView)
    }
    
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    override func layoutSubviews() {
        super.layoutSubviews()
        
        // 调整UIImageView的布局
        customImageView.frame = CGRect(x: 0, y: 0, width: contentView.frame.width, height: contentView.frame.height)
        
        // 根据UIImageView的高度调整其他子视图的布局
        // ...
    }
}

class ViewController: UIViewController, UITableViewDelegate, UITableViewDataSource {
    var tableView: UITableView!
    var data: [UIImage] = [] // 存储图片数据的数组
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        tableView = UITableView(frame: view.bounds, style: .plain)
        tableView.delegate = self
        tableView.dataSource = self
        tableView.register(CustomTableViewCell.self, forCellReuseIdentifier: "CustomCell")
        view.addSubview(tableView)
    }
    
    // 返回表单元格的高度
    func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
        let image = data[indexPath.row]
        let aspectRatio = image.size.width / image.size.height
        let cellWidth = tableView.frame.width
        let cellHeight = cellWidth / aspectRatio
        
        return cellHeight
    }
    
    // 返回表单元格的数量
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return data.count
    }
    
    // 返回表单元格的内容
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "CustomCell", for: indexPath) as! CustomTableViewCell
        
        let image = data[indexPath.row]
        cell.customImageView.image = image
        
        return cell
    }
}