Question

假设我有一个整数序列，需要按如下子序列进行拆分：

def splitBySeq(xs: Seq[Int], ys: Seq[Int]): (Seq[Int], Seq[Int]) = ???

val xs = List(1, 2, 3, 4, 5)
splitBySeq(xs, Nil)                 // (List(1, 2, 3, 4, 5), Nil)
splitBySeq(xs, List(1))             // (Nil, List(2, 3, 4, 5))
splitBySeq(xs, List(5))             // (List(1, 2, 3, 4), Nil)
splitBySeq(xs, List(3, 4))          // (List(1, 2), List(5))
splitBySeq(xs, List(11, 12))        // (List(1, 2, 3, 4, 5), Nil)
splitBySeq(xs, List(1, 2, 3, 4, 5)) // (Nil, Nil)

如果ys是xs的子序列，则该函数应返回一对序列-xs1和xs2，以便xs1 ++ ys ++ xs2 == xs。否则，该函数将返回xs。

您将如何实现splitBySeq？

Answer 1

这似乎可以满足您的需求。

def splitBySeq(xs: Seq[Int], ys: Seq[Int]): (Seq[Int], Seq[Int]) = {
  val idx = xs indexOfSlice ys
  if (idx < 0) (xs, Nil)
  else {
    val (a,b) = xs splitAt idx
    (a, b drop ys.length)
  }
}

请注意，在第一个测试用例splitBySeq(xs, Nil)中，由于Nil与任何Seq的零索引匹配，结果序列被切换。

Answer 2

另一种解决方案，使用尾递归函数进行一次迭代：

def splitBySeq[A](xs: Seq[A], ys: Seq[A]): (Seq[A], Seq[A]) = {

  @tailrec
  def go(a: List[A], b: List[A], acc: List[A], rest: List[A]): (Seq[A], Seq[A]) = {

    (a, b) match {
      case (z :: zs, w :: ws)  => {
        if(z == w) {
          go(zs, ws, z :: acc, rest)
        } else{
          go(zs, ys.toList, Nil, z :: (acc ++ rest))
        }
      }
      case (zs, Nil) => (rest.reverse, zs)
      case (Nil, _) => (rest.reverse, Nil)
    }
  }

  if(ys.isEmpty) {
    (xs, Nil)
  } else {
    go(xs.toList, ys.toList, Nil, Nil)
  }

}

如何用子序列分割序列？

2 个答案: