根据单元格数量调整UICollectionView的大小

Question

我有一个包含1-4个单元格的UICollection视图，UITextField下方的视图被限制在其底部。

我希望能够调整UICollectionView的大小，所以如果它有1个或2个单元格，UICollectionView的高度就是这个高度，否则假设它有3个或4个单元格，则高度为2 *单元格高度+垂直间距。

在没有所有数学的情况下，有没有一种简单的方法可以做到这一点？拥有约束系统似乎很奇怪，并且仍然必须使用这种费力的数学来确定UIView的大小。

一如既往地谢谢