scala - 有条件地合并列表元素

Question

我想根据一些表达式有条件地合并列表中彼此后面的元素。
一个更好地解释我想做的事情的例子:

来自:
val list = List("a1", "a2", "b1", "b2", "b3", "a3", "a4")
我想将所有以 b 开头的元素合并为一个元素，以获得这样的结果列表:

List("a1", "a2", "b1-b2-b3", "a3", "a4")

在我的用例中，b 元素总是按顺序排列，但 b 元素的数量可以从没有元素到数十个元素不等。

我试过做类似的事情

list.foldLeft("")((s1, s2) => if (s1.matches("""b\d""") && s2.matches("""b\d""")) s1 + "-" + s2 else s1)

但它不会使我有任何用处。

关于如何解决这个问题的任何建议？

Answer 1

可以通过 foldLeft 来完成并查看列表中最近插入的元素:

list.foldLeft(List[String]()) {
  case (Nil, str) => str :: Nil
  case (head :: tail, str) =>
    if (head(0) == 'b' && str(0) == 'b') (s"$head-$str") :: tail
    else str :: head :: tail
}.reverse 
//> res0: List[String] = List(a1, a2, b1-b2-b3, a3, a4)

模式匹配也可以重写(如果你觉得更清楚):

list.foldLeft(List[String]()) {
  case (head :: tail, str) if (head(0) == 'b' && str(0) == 'b') =>
    (s"$head-$str") :: tail
  case (other, str) =>
    str :: other
}.reverse