homework_9 - Тебайкин Максим

homeworks/homework_9/src/main/scala/ListOps.scala

@@ -1,3 +1,6 @@
+import DataList.{EmptyList, NonEmptyList}
+
+import scala.:+
 import scala.annotation.tailrec
 
 object ListOps {
@@ -9,7 +12,13 @@ object ListOps {
    * @param f функция свёртывания. Применяется попарно к предыдущему результату применения и i-ому элементу списка
    * @return None - если список пустой
    */
-  def foldOption[T](f: (T, T) => T): DataList[T] => Option[T] = ???
+  def foldOption[T](f: (T, T) => T): DataList[T] => Option[T] = {
+    case NonEmptyList(head, tail) => foldOption(f)(tail) match {
+      case Some(value) => Some(f(head, value))
+      case None => Some(head)
+    }
+    case EmptyList => None
+  }
 
 
   /**
@@ -21,16 +30,26 @@ object ListOps {
      * Используйте для суммирования двух чисел любого типа (Int, Long, Double, Float etc)
      */
     def sumT(a: T, b: T) = implicitly[Numeric[T]].plus(a, b)
+    foldOption(sumT)(list).getOrElse(Numeric[T].zero)
+  }
 
-    ???
+  @tailrec
+  def reverse[T](l: DataList[T], buffer: DataList[T]): DataList[T] = l match {
+    case NonEmptyList(head, tail) => reverse(tail, NonEmptyList(head, buffer))
+    case EmptyList => buffer
   }
 
   /**
    * Фильтрация списка. Хвостовая рекурсия
+   *
    * @param f - фильтрующее правило (если f(a[i]) == true, то элемент остаётся в списке)
    */
   @tailrec
-  private def filterImpl[T](f: T => Boolean)(buffer: DataList[T])(l: DataList[T]): DataList[T] = ???
+  private def filterImpl[T](f: T => Boolean)(buffer: DataList[T])(l: DataList[T]): DataList[T] = l match {
+    case EmptyList => reverse(buffer, EmptyList)
+    case NonEmptyList(head, tail) =>
+      filterImpl(f)(if (f(head)) NonEmptyList(head, buffer) else buffer)(tail)
+  }
 
   final def filter[T](f: T => Boolean): DataList[T] => DataList[T] = filterImpl(f)(DataList.EmptyList)
 
@@ -43,6 +62,6 @@ object ListOps {
    * Используя композицию функций реализуйте collect. Collect - комбинация filter и map.
    * В качестве фильтрующего правила нужно использовать f.isDefinedAt
    */
-  def collect[A, B](f: PartialFunction[A, B]): DataList[A] => DataList[B] = ???
+  def collect[A, B](f: PartialFunction[A, B]): DataList[A] => DataList[B] = filter(f.isDefinedAt).andThen(map(f))
 
 }