Scala - выбор элементов из упорядоченного списка

Question

Я ищу хороший способ удалить первые N элементов, которые равны из упорядоченного списка, например.

List(1,1,1,2,3,3) 

должен вернуть

removeSame(list) -> (1,1,1) 

Есть хороший способ сделать это, а не удалить голову списка, а затем использовать takeWhile на остаток, и, наконец, с помощью dropwhile? Я могу думать о простом нефункциональном решении, но мне было интересно, существует ли какой-либо функциональный

Answer 1

Функциональный способ избежать дублирования takeWhile и dropWhile, чтобы получить префикс и остаток использует span, т.е.

scala> val list = List(1,1,1,2,3,3) 
list: List[Int] = List(1, 1, 1, 2, 3, 3) 

scala> val (prefix, rest) = list span (_ == list.head) 
prefix: List[Int] = List(1, 1, 1) 
rest: List[Int] = List(2, 3, 3) 

Answer 2

Это то, что вы ищете?

scala> val l = List(1,1,1,2,3,3) 
l: List[Int] = List(1, 1, 1, 2, 3, 3) 

scala> l.takeWhile(_ == l.head) 
res6: List[Int] = List(1, 1, 1) 

scala> val l = List() 
l: List[Nothing] = List() 

scala> l.takeWhile(_ == l.head) 
res7: List[Nothing] = List() 

Answer 3

Не самый лучший способ сделать это, но это также работает:

def removeSame(l: List) = if(!l.isEmpty) l.groupBy(x => x)(l.head) else List()