scala – Group List元素,距离小于x

发布时间：2020-12-16 09:54:59 所属栏目：安全来源：网络整理

导读：我试图找出一种方法来根据元素之间的x距离对列表中的所有对象进行分组. 例如,如果距离是1那么 List(2,3,1,6,10,7,11,12,14) 会给 List(List(1,2,3),List(6,7),List(10,12),List(14)) 我只能提出棘手的方法和循环,但我想必须有一个更清洁的解决方案. 解决方法

我试图找出一种方法来根据元素之间的x距离对列表中的所有对象进行分组.

例如,如果距离是1那么

List(2,3,1,6,10,7,11,12,14)

会给

List(List(1,2,3),List(6,7),List(10,12),List(14))

我只能提出棘手的方法和循环,但我想必须有一个更清洁的解决方案.

解决方法

您可以尝试对列表进行排序,然后在其上使用foldLeft.基本上是这样的

def sort = {
    val l = List(2,14)
    val dist = 1
    l.sorted.foldLeft(List(List.empty[Int]))((list,n) => {
      val last = list.head
      last match {
        case h::q  if Math.abs(last.head-n) > dist=> List(n) :: list
        case _ => (n :: last ) :: list.tail 
      }
    }
    )
  }

结果似乎没问题但反过来了.如果需要,在需要时在列表上调用“反向”.代码变成了

val l = List(2,14)
    val dist = 1
    val res = l.sorted.foldLeft(List(List.empty[Int]))((list,n) => {
       val last = list.head
       last match {
         case h::q  if Math.abs(last.head-n) > dist=> List(n) :: (last.reverse :: list.tail)
        case _ => (n :: last ) :: list.tail
      }
    }
).reverse

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!