在collectionview中以不同大小的网格格式显示图像

Question

我想在集合视图中以网格格式显示带有加号（即+ 2，+ 3等）的各种大小的图像。看起来就像...

我知道使用UICollectionViewFlowLayout可以实现。因此，我提到了this堆栈溢出问题，并尝试添加此行代码。但这没用

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {

        switch indexPath.item {
        case 0,1:
            return CGSize(width: (UIScreen.main.bounds.width - 16) / 2, height: (UIScreen.main.bounds.width - 16) / 2)
        default:
            return CGSize(width: (UIScreen.main.bounds.width - 32) / 3, height:  (UIScreen.main.bounds.width) / 3)
        }
    }

此外，如果选择的图像数量是2或3，则这些图像的排列方式应如此。

如何使图像看起来像这样...？