查找两个列表中的常见项(Scala)

时间:2016-09-22 15:49:56

标签: scala scala-collections

新手Scala问题。

我正在尝试实现一个接收两个列表的函数,找到一个公共项目,而不是进行操作并创建一个新列表

我有一个案例类

case class weightedFruits(fruits: Set[String], weight: Double) 

以及两个列表weightedFruitsListfilteredWeightedFruitsList

// set is sorted 
val weightedFruitsList = List(
weightedFruits(Set("banana"), 200),
weightedFruits(Set("banana", "orange"), 180),
weightedFruits(Set("banana", "orange", "apple"), 170),
weightedFruits(Set("feijoa", "fig"), 201))


 //filtered List , Set sorted, contains "melon" as last member
val filteredWeightedFruitsList = List(
weightedFruits(Set("banana", "melon"), 250),
weightedFruits(Set("banana", "orange", "melon"), 270),
weightedFruits(Set("banana", "orange", "apple", "melon"), 365))

我想查看filteredWeightedFruitsList中的每个项目,在weightedFruitsList中找到相同的项目,进行小操作并创建新的List[weightedFruits]

我的(还没有工作)代码:

def conf :Option[List[weightedFruits]] = {

  for (filtered <- filteredWeightedFruitsList){

     weightedFruitsList.find{

      case x if ( x.fruits equals filtered.fruits.dropRight(1) ) => return weightedFruits(x.fruits, x.weight / filtered.weight)]
      case _ => false
   }    
  }    
}

使用此代码我有两个问题: 1)type mismatch; found : Unit required: Option

2)type mismatch; found : weightedFruits required: Option[List[weightedFruits]]

欢迎任何想法。对不起,如果我的问题让你发疯...

最后一个问题是否有更有效的方法来完成这项任务?

由于

1 个答案:

答案 0 :(得分:2)

type mismatch; found : weightedFruits required: Option[List[weightedFruits]]是由conf方法未返回Option[List[weightedFruits]]类型结果造成的。也许你可以尝试使用for yield来做到这一点。

def conf :List[weightedFruits] = for {
   f <- filteredWeightedFruitsList
   t <- weightedFruitsList.find(i => f.fruits.dropRight(1) == i.fruits)
} yield t.copy(weight = t.weight / f.weight)

copy方法将复制case class,并使用name覆盖某些字段