Digamos que tengo una lista de enteros: [2, 4, 4, 6, 1, 8, 3, 5, 2, 2, 2] , y quiero dividir esta lista en sublistas para que cada sublista contenga valores que coincida con el predicado, o si no lo hace, solo el valor que no coincidió, y debe estar en el orden de la lista original.
Entonces, por ejemplo, si el predicado es "es par", espero obtener el siguiente resultado: [[2, 4, 4, 6],[1],[8],[3, 5],[2, 2, 2]]
Tengo un código que funciona, pero en mi opinión es bastante feo y siento que debería haber una forma mucho mejor de escribirlo:
private fun <T : Any> toSubLists(values: List<T>, predicate: (T) -> Boolean): List<List<T>> { val subLists = mutableListOf<List<T>>() var currentList = mutableListOf<T>() values.forEach { if (!predicate(it)) { if (currentList.isNotEmpty()) { subLists.add(currentList) currentList = mutableListOf() } subLists.add(listOf(it)) } else { currentList.add(it) } } if (currentList.isNotEmpty()) { subLists.add(currentList) } return subLists }No estoy seguro si es mejor porque puede ser más difícil de entender, pero esta es una forma de hacerlo.
private fun <T : Any> toSubLists(values: List<T>, predicate: (T) -> Boolean): List<List<T>> = values.withIndex().groupBy({ var index = it.index while (index != 0 && predicate(values[index]) && predicate(values[index - 1])) index-- index }){ it.value }.values.toList()míralo funcionando aquí
Hay una solicitud de función para una versión genérica de esto que acepta otros selectores de clave además de los predicados.
Mientras tanto, para su caso de uso, creo que podemos simplificar un poco:
fun <T> List<T>.groupRunsBy(predicate: (T) -> Boolean): List<List<T>> { val result = mutableListOf<MutableList<T>>() var currentlyMatching: Boolean? = null forEach { val itemMatches = predicate(it) if (itemMatches == currentlyMatching) { result.last().add(it) } else { currentlyMatching = itemMatches result.add(mutableListOf(it)) } } return result } fun main() { val list = listOf(2, 4, 4, 6, 1, 8, 3, 5, 2, 2, 2) println(list.groupRunsBy { it % 2 == 0 }) // [[2, 4, 4, 6], [1], [8], [3, 5], [2, 2, 2]] }Probado aquí: https://pl.kotl.in/OHVr7S6lX
Esta versión se puede cambiar fácilmente para admitir cualquier selector que devuelva claves no nulas.
fun <T : Any> toSubLists(values: List<T>, predicate: (T) -> Boolean): List<List<T>> { val items = values.map { predicate(it) to listOf(it) }.toMutableList() for (i in items.indices.last downTo 1) { if (items[i - 1].first && items[i].first) { items[i - 1] = items[i - 1].first to items[i - 1].second + items[i].second items.removeAt(i) } } return items.map { it.second } } val list1 = listOf(2, 4, 4, 6, 1, 8, 3, 5, 2, 2, 2) val isOdd: (Int) -> Boolean = { it % 2 != 0 } val isEven: (Int) -> Boolean = { it % 2 == 0 } println(toSubLists(list1, isOdd)) // [[2], [4], [4], [6], [1], [8], [3, 5], [2], [2], [2]] println(toSubLists(list1, isEven)) // [[2, 4, 4, 6], [1], [8], [3], [5], [2, 2, 2]]