Определение элемента, по которому был выполнен клик можно оптимизировать. Например, можно не проходить по всему списку и не считать позицию каждого элемента, а преобразовать Y позицию клика в Y позицию в LazyColumn, а затем найти нужный элемент путем бинарного поиска
Создаём iOS-style Wheel Picker в Jetpack Compose. Часть 1