animated-vertical-grid-5.kt

var itemsOffsets by remember { mutableStateOf(mapOf<KEY, ItemOffset>()) }
key(itemKeys) {
   itemsOffsets = items.mapIndexed { index, item ->
       val key = itemKey(item)
       key to when {
           itemsOffsets.containsKey(key) -> itemsOffsets.getValue(key)
           else -> ItemOffset(gridOffsets[index])
       }
   }.toMap()
}