iOS设备屏幕尺寸

Question

我想根据设备的高度分配表格视图单元格的高度。 我知道我可以获取高度，但是可以像表格视图单元格高度一样分配

func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
    switch indexPath.section {
      case 0:
        return 0.9 * UIScreen.main.bounds.width + 440 /// i want like lessthan(0.9 * UIScreen.main.bounds.width + 440)
      case 1:
        let episodesCount: CGFloat = 2
        let seasonDetailsHeight: CGFloat = 104
        return 100 * episodesCount + seasonDetailsHeight
      case 2:
        return ((UIScreen.main.bounds.height) / 4)
      default: return 0
    }
  }