Операции со списками

Линейный поиск

В любых списках вы можете найти позицию элемента, используя функции indexOf() и lastIndexOf(). Они возвращают первую и последнюю позицию элемента, равного заданному аргументу в списке. Если таких элементов нет, обе функции возвращают -1.

fun main() {
//sampleStart
    val numbers = listOf(1, 2, 3, 4, 2, 5)
    println(numbers.indexOf(2))
    println(numbers.lastIndexOf(2))
//sampleEnd
}

Также есть пара функций, которые принимают предикат и ищут элементы, соответствующие ему:

• indexOfFirst() возвращает индекс первого элемента, соответствующего предикату, или -1 если таких элементов нет.

• indexOfLast() возвращает индекс последнего элемента, соответствующего предикату, или -1 если таких элементов нет.

fun main() {
//sampleStart
    val numbers = mutableListOf(1, 2, 3, 4)
    println(numbers.indexOfFirst { it > 2})
    println(numbers.indexOfLast { it % 2 == 1})
//sampleEnd
}

Бинарный поиск в отсортированных списках

Есть еще один способ поиска элементов в списках — бинарный поиск. Он работает значительно быстрее, чем другие встроенные функции поиска, но требует, чтобы список был отсортирован по возрастанию в соответствии с определенным порядком: естественным или другим, заданным в параметре функции. В противном случае результат не определен.

Чтобы найти элемент в отсортированном списке, вызовите функцию binarySearch(), передав значение в качестве аргумента. Если такой элемент существует, функция возвращает его индекс; в противном случае она возвращает (-insertionPoint - 1), где insertionPoint — индекс, куда следует вставить этот элемент, чтобы список оставался отсортированным. Если существует более одного элемента с заданным значением, поиск может вернуть любой из их индексов.

Вы также можете указать диапазон индексов для поиска: в этом случае функция ищет только между двумя указанными индексами.

fun main() {
//sampleStart
    val numbers = mutableListOf("one", "two", "three", "four")
    numbers.sort()
    println(numbers)
    println(numbers.binarySearch("two"))  // 3
    println(numbers.binarySearch("z")) // -5
    println(numbers.binarySearch("two", 0, 2))  // -3
//sampleEnd
}

data class Product(val name: String, val price: Double)

fun main() {
//sampleStart
    val productList = listOf(
        Product("WebStorm", 49.0),
        Product("AppCode", 99.0),
        Product("DotTrace", 129.0),
        Product("ReSharper", 149.0))

    println(productList.binarySearch(Product("AppCode", 99.0), compareBy<Product> { it.price }.thenBy { it.name }))
//sampleEnd
}

fun main() {
//sampleStart
    val colors = listOf("Blue", "green", "ORANGE", "Red", "yellow")
    println(colors.binarySearch("RED", String.CASE_INSENSITIVE_ORDER)) // 3
//sampleEnd
}

import kotlin.math.sign
//sampleStart
data class Product(val name: String, val price: Double)

fun priceComparison(product: Product, price: Double) = sign(product.price - price).toInt()

fun main() {
    val productList = listOf(
        Product("WebStorm", 49.0),
        Product("AppCode", 99.0),
        Product("DotTrace", 129.0),
        Product("ReSharper", 149.0))

    println(productList.binarySearch { priceComparison(it, 99.0) })
}
//sampleEnd

Добавление

Для добавления элементов в определённую позицию в списке используйте add() и addAll(), передавая позицию вставки элемента в качестве дополнительного аргумента. Все элементы, следующие за указанной позицией, сдвигаются вправо.

fun main() {
//sampleStart
    val numbers = mutableListOf("one", "five", "six")
    numbers.add(1, "two")
    numbers.addAll(2, listOf("three", "four"))
    println(numbers)
//sampleEnd
}

Обновление

Списки также предлагают функцию замены элемента в заданной позиции - set() и её операторную форму []. set() не изменяет индексы других элементов.

fun main() {
//sampleStart
    val numbers = mutableListOf("one", "five", "three")
    numbers[1] =  "two"
    println(numbers)
//sampleEnd
}

fill() просто заменяет все элементы коллекции указанным значением.

fun main() {
//sampleStart
    val numbers = mutableListOf(1, 2, 3, 4)
    numbers.fill(3)
    println(numbers)
//sampleEnd
}

Удаление

Для удаления элемента по заданной позиции из списка используйте функцию removeAt(), передав позицию в качестве аргумента. Все индексы элементов, следующих за удалённым элементом, уменьшатся на единицу.

fun main() {
//sampleStart
    val numbers = mutableListOf(1, 2, 3, 4, 3)    
    numbers.removeAt(1)
    println(numbers)
//sampleEnd
}

Сортировка

В разделе Порядок элементов коллекции мы описываем операции, которые извлекают элементы коллекции в определённом порядке. Для изменяемых списков стандартная библиотека предлагает аналогичные расширяющие функции, которые выполняют те же операции упорядочения на месте. При применении такой операции к экземпляру списка порядок элементов в этом конкретном экземпляре изменяется.

Функции сортировки на месте имеют сходные имена с функциями, применяемыми к неизменяемым спискам, но без суффикса ed/d:

• sort* вместо sorted* в названиях всех функций сортировки: sort(), sortDescending(), sortBy() и т.д.

• shuffle() вместо shuffled().

• reverse() вместо reversed().

asReversed(), вызываемая для изменяемого списка, возвращает другой изменяемый список, который является обратным представлением исходного списка. Изменения в этом представлении отражаются в исходном списке. Приведённый ниже пример демонстрирует функции сортировки для изменяемых списков:

fun main() {
//sampleStart
    val numbers = mutableListOf("one", "two", "three", "four")

    numbers.sort()
    println("Sort into ascending: $numbers")
    numbers.sortDescending()
    println("Sort into descending: $numbers")

    numbers.sortBy { it.length }
    println("Sort into ascending by length: $numbers")
    numbers.sortByDescending { it.last() }
    println("Sort into descending by the last letter: $numbers")
    
    numbers.sortWith(compareBy<String> { it.length }.thenBy { it })
    println("Sort by Comparator: $numbers")

    numbers.shuffle()
    println("Shuffle: $numbers")

    numbers.reverse()
    println("Reverse: $numbers")
//sampleEnd
}