Kotlin-4 homework Collections

src/main/kotlin/ru/otus/homework/NaturalList.kt

@@ -1,9 +1,5 @@
 package ru.otus.homework
 
-/**
- * Список натуральных чисел от 1 до n
- * @param n Последнее натуральное число в списке
- */
 class NaturalList(n: Int) : List<Int> {
     override val size: Int = n
 
@@ -31,35 +27,19 @@ class NaturalList(n: Int) : List<Int> {
 
     override fun lastIndexOf(element: Int): Int = indexOf(element)
 
-    /**
-     * Вернуть под-список этого списка, включая [fromIndex] и НЕ включая [toIndex]
-     */
-    override fun subList(fromIndex: Int, toIndex: Int): List<Int> {
-        TODO("Not yet implemented")
-    }
+    override fun subList(fromIndex: Int, toIndex: Int): List<Int> =
+        this.toList().slice(fromIndex until toIndex)
 
-    /**
-     * Returns true if list contains all numbers in the collection
-     */
-    override fun containsAll(elements: Collection<Int>): Boolean {
-        TODO("Not yet implemented")
-    }
+    override fun containsAll(elements: Collection<Int>): Boolean =
+        this.toList().containsAll(elements)
 
     override fun toString(): String {
         return "NaturalList(1..$size)"
     }
 
-    /**
-     * Функция должна возвращать true, если сравнивается с другой реализацией списка тех же чисел
-     * Например, NaturalList(5) должен быть равен listOf(1,2,3,4,5)
-     */
-    override fun equals(other: Any?): Boolean = false
-
-    /**
-     * Функция должна возвращать тот же hash-code, что и список другой реализации тех же чисел
-     * Например, NaturalList(5).hashCode() должен быть равен listOf(1,2,3,4,5).hashCode()
-     */
-    override fun hashCode(): Int = -1
+    override fun equals(other: Any?): Boolean = this.toList() == other
+
+    override fun hashCode(): Int = this.toList().hashCode()
 }
 
 private class NaturalIterator(private val n: Int) : Iterator<Int> {
@@ -73,19 +53,21 @@ private class NaturalIterator(private val n: Int) : Iterator<Int> {
 }
 
 private class NaturalListIterator(private val n: Int, index: Int = 0) : ListIterator<Int> {
-    private var index:Int = index.coerceIn(0, n - 1)
+    private var index: Int = index.coerceIn(0, n - 1)
     override fun hasNext(): Boolean = index < n
     override fun hasPrevious(): Boolean = index > 0
     override fun next(): Int = if (hasNext()) {
         ++index
     } else {
         throw NoSuchElementException()
     }
+
     override fun nextIndex(): Int = index
     override fun previous(): Int = if (hasPrevious()) {
         index--
     } else {
         throw NoSuchElementException()
     }
+
     override fun previousIndex(): Int = index
 }

src/main/kotlin/ru/otus/homework/mapswap/mapSwap.kt

@@ -1,6 +1,10 @@
 package ru.otus.homework.mapswap
 
-/**
- * Меняет местами ключи и значения
- */
-fun <K, V> Map<K, V>.swap(): Map<V, K> = TODO("Доделать swap")
+fun <K, V> Map<K, V>.swap(): Map<V, K> {
+    val invertedMap = mutableMapOf<V, K>()
+
+    this.forEach {
+        invertedMap[it.value] = it.key
+    }
+    return invertedMap
+}

src/main/kotlin/ru/otus/homework/persons/persons.kt

@@ -1,13 +1,8 @@
 package ru.otus.homework.persons
 
-/**
- * Отсортировать список персон по возрасту в порядке убывания
- */
-fun List<Person>.sortByAge(): List<Person> = TODO("Доделать sortByAge")
-
-/**
- * Отсортировать список персон по фамилии
- * - Фамилии сортируются по алфавиту в порядке возрастания
- * - Если фамилии совпадают, персоны сортируются по имени в порядке возрастания
- */
-fun List<Person>.sortByName(): List<Person> = TODO("Доделать sortBySurname")
+fun List<Person>.sortByAge(): List<Person> = this.sortedByDescending { it.age }
+
+fun List<Person>.sortByName(): List<Person> =
+    this.sortedWith(compareBy<Person> { it.surname }.thenBy { it.name })
+
+

src/test/kotlin/ru/otus/homework/NaturalListTest.kt

@@ -3,7 +3,7 @@ package ru.otus.homework
 import org.junit.jupiter.api.Assertions.*
 import org.junit.jupiter.api.Test
 
-    class NaturalListTest {
+class NaturalListTest {
 
     @Test
     fun returnsRequestedSize() {