如何根據階

Question

謂語dropFirst takeFirst項目我有這些元素在列表中

val list = (1,2,3,0, 0)

我需要這個名單分成兩個列表洗牌。

val list1 = (1) 
val list2 = (2, 3, 0, 0) 

第一個列表包含一個項目是不是0，第二個列表包含項目的其餘部分。

修復下面的代碼，所以list1 never contains 0和list1 :: list2 == list

val list = (1, 2, 0, 3, 0).shuffle 

val list1 = list(0) 
val list2 = list drop 1 

有點類似於此：How should I remove the first occurrence of an object from a list in Scala?

Answer 1

我會從以下功能開始：

def sep(l: List[Int]) = 
    (l.find(_ != 0).toList, 
    l.takeWhile(_ == 0) ::: l.dropWhile(_ == 0).tail) 

它返回你所需要的，如果解決方案存在：

sep(List(0, 1, 0, 2, 3)) -> (List(1),List(0, 0, 2, 3)) 
sep(List(1, 4, 0, 2, 3)) -> (List(1),List(4, 0, 2, 3)) 
sep(List(0, 0, 4, 5)) -> (List(4),List(0, 0, 5)) 

但是，如果您有List(0,0) ...下一步是遞歸編寫它，因此它只使用一次迭代。

Answer 2

許多這樣做的方法，這裏是我的：

val list = List(0,1,0,3,2) 
val list1 = List(list.find(_!=0).get) 
val list2 = list.filter(_!=list1(0)) 

最終結果（斯卡拉REPL）

list: List[Int] = List(0, 1, 0, 3, 2) 
list1: List[Int] = List(1) 
list2: List[Int] = List(0, 0, 3, 2) 

請注意，此代碼假定list具有在列表中的一個非零成員。

更新（每@ theArchetypalPaul此言）支持列表與非唯一非零成員：

val temp = list.span(_ == 0) 
val list1 = List(temp._2.head) 
val list2 = temp._1 ++ temp._2.tail