Что такое идиоматический способ Scala "удалить" один элемент из неизменяемого списка?
У меня есть список, который может содержать элементы, которые будут сравнивать как равные. Я хотел бы аналогичный список, но с одним удаленным элементом. Поэтому из (A, B, C, B, D) Я хотел бы иметь возможность "удалить" только один B, чтобы получить, например (A, C, B, D). Порядок элементов в результате не имеет значения.
У меня есть рабочий код, написанный в стиле Lisp в Scala. Есть ли более идиоматические способ чтобы сделать это?
контекст-это карточная игра, где две колоды стандартных карт в игре, так что может будьте дубликаты карт, но по-прежнему играл по одному за раз.
def removeOne(c: Card, left: List[Card], right: List[Card]): List[Card] = {
if (Nil == right) {
return left
}
if (c == right.head) {
return left ::: right.tail
}
return removeOne(c, right.head :: left, right.tail)
}
def removeCard(c: Card, cards: List[Card]): List[Card] = {
return removeOne(c, Nil, cards)
}
12 ответов:
Я не видел такой возможности в ответах выше, так что:
scala> def remove(num: Int, list: List[Int]) = list diff List(num) remove: (num: Int,list: List[Int])List[Int] scala> remove(2,List(1,2,3,4,5)) res2: List[Int] = List(1, 3, 4, 5)
Edit:
scala> remove(2,List(2,2,2)) res0: List[Int] = List(2, 2)
Как Шарм :-).
вы можете использовать метод filterNot.
val data = "test" list = List("this", "is", "a", "test") list.filterNot(elm => elm == data)
вы можете попробовать это:
scala> val (left,right) = List(1,2,3,2,4).span(_ != 2) left: List[Int] = List(1) right: List[Int] = List(2, 3, 2, 4) scala> left ::: right.tail res7: List[Int] = List(1, 3, 2, 4)
и как метод:
def removeInt(i: Int, li: List[Int]) = { val (left, right) = li.span(_ != i) left ::: right.drop(1) }
к сожалению, иерархия коллекций попала в небольшую путаницу с
-
onList
. ИбоArrayBuffer
он работает так же, как вы могли бы надеяться:scala> collection.mutable.ArrayBuffer(1,2,3,2,4) - 2 res0: scala.collection.mutable.ArrayBuffer[Int] = ArrayBuffer(1, 3, 2, 4)
но, к сожалению,
List
в конечном итоге сfilterNot
-реализация стиля и, таким образом, делает "неправильную вещь" и бросает предупреждение об осуждении на вас (достаточно разумно, так как это на самом делеfilterNot
ing):scala> List(1,2,3,2,4) - 2 warning: there were deprecation warnings; re-run with -deprecation for details res1: List[Int] = List(1, 3, 4)
так, пожалуй, проще всего конвертировать
List
в a коллекция, которая делает это правильно, а затем снова конвертирует:import collection.mutable.ArrayBuffer._ scala> ((ArrayBuffer() ++ List(1,2,3,2,4)) - 2).toList res2: List[Int] = List(1, 3, 2, 4)
кроме того, вы можете сохранить логику кода, который у вас есть, но сделать стиль более идиоматичным:
def removeInt(i: Int, li: List[Int]) = { def removeOne(i: Int, left: List[Int], right: List[Int]): List[Int] = right match { case r :: rest => if (r == i) left.reverse ::: rest else removeOne(i, r :: left, rest) case Nil => left.reverse } removeOne(i, Nil, li) } scala> removeInt(2, List(1,2,3,2,4)) res3: List[Int] = List(1, 3, 2, 4)
def removeAtIdx[T](idx: Int, listToRemoveFrom: List[T]): List[T] = { assert(listToRemoveFrom.length > idx && idx >= 0) val (left, _ :: right) = listToRemoveFrom.splitAt(idx) left ++ right }
// throws a MatchError exception if i isn't found in li def remove[A](i:A, li:List[A]) = { val (head,_::tail) = li.span(i != _) head ::: tail }
в качестве одного из возможных решений можно найти индекс первого подходящего элемента, а затем удалить элемент по этому индексу:
def removeOne(l: List[Card], c: Card) = l indexOf c match { case -1 => l case n => (l take n) ++ (l drop (n + 1)) }
как о
def removeCard(c: Card, cards: List[Card]) = { val (head, tail) = cards span {c!=} head ::: (tail match { case x :: xs => xs case Nil => Nil }) }
Если вы видите
return
, что-то не так.
просто еще одна мысль о том, как это сделать с помощью складки:
def remove[A](item : A, lst : List[A]) : List[A] = { lst.:\[List[A]](Nil)((lst, lstItem) => if (lstItem == item) lst else lstItem::lst ) }
Общее Решение Хвостовой Рекурсии:
def removeElement[T](list: List[T], ele: T): List[T] = { @tailrec def removeElementHelper(list: List[T], accumList: List[T] = List[T]()): List[T] = { if (list.length == 1) { if (list.head == ele) accumList.reverse else accumList.reverse ::: list } else { list match { case head :: tail if (head != ele) => removeElementHelper(tail, head :: accumList) case head :: tail if (head == ele) => (accumList.reverse ::: tail) case _ => accumList } } } removeElementHelper(list) }
val list : Array[Int] = Array(6, 5, 3, 1, 8, 7, 2) val test2 = list.splitAt(list.length / 2)._2 val res = test2.patch(1, Nil, 1)
object HelloWorld { def main(args: Array[String]) { var months: List[String] = List("December","November","October","September","August", "July","June","May","April","March","February","January") println("Deleting the reverse list one by one") var i = 0 while (i < (months.length)){ println("Deleting "+months.apply(i)) months = (months.drop(1)) } println(months) } }