第十章 Scala 容器基础(二十一):从集合中提取不重复的元素

点点圈 提交于 2020-04-07 06:02:48

Problem

    你有一个集合,内部有很多重复元素,你想要把这些重复的元素只保留一份。

Solution

    使用Distinct方法:

scala> val x = Vector(1, 1, 2, 3, 3, 4)
x: scala.collection.immutable.Vector[Int] = Vector(1, 1, 2, 3, 3, 4)

scala> val y = x.distinct
y: scala.collection.immutable.Vector[Int] = Vector(1, 2, 3, 4)

    这个distinct方法返回一个新的集合,重复元素只保留一份。记得使用一个新的变量来指向这个新的集合,无论你使用的是mutable集合还是immutable集合。

    如果你突然需要一个set,那么直接吧你的集合转化成为一个set也是去掉重复元素的方式:

scala> val s = x.toSet
s: scala.collection.immutable.Set[Int] = Set(1, 2, 3, 4)

    因为Set对于一样的元素只能保存一份,所以把Array,List,Vector或者其他的集合转化成Set可以去掉重复元素。实际上这就是distinct方法的工作远离。Distinct方法的源代码显示了他就是实用了一个mutable.HashSet的实例。

Using distinct with your own classes

    要想对你自己定义的集合元素类型使用distinct方法,你需要实现equals和hashCode方法。举个例子,下面这个类就可以使用disticnt方法,因为我们实现了这两个方法:

class Person(firstName: String, lastName: String) {
  override def toString = s"$firstName $lastName"
  def canEqual(a: Any) = a.isInstanceOf[Person]
  override def equals(that: Any): Boolean = {
    that match {
      case that: Person => that.canEqual(this) && this.hashCode == that.hashCode
      case _ => false
    }
  }
  override def hashCode: Int = {
    val prime = 31
    var result = 1
    result = prime * result + lastName.hashCode
    result = prime * result + (if(firstName == null) 0 else firstName.hashCode)
    return result
  }
}

scala> class Person(firstName: String, lastName: String) {
     |   override def toString = s"$firstName $lastName"
     |   def canEqual(a: Any) = a.isInstanceOf[Person]
     |   override def equals(that: Any): Boolean = {
     |     that match {
     |       case that: Person => that.canEqual(this) && this.hashCode == that.hashCode
     |       case _ => false
     |     }
     |   }
     |   override def hashCode: Int = {
     |     val prime = 31
     |     var result = 1
     |     result = prime * result + lastName.hashCode
     |     result = prime * result + (if(firstName == null) 0 else firstName.hashCode)
     |     return result
     |   }
     | }
defined class Person

object Person {
  def apply(firstName: String, lastName: String) = new Person(firstName, lastName)
}

scala> object Person {
     |   def apply(firstName: String, lastName: String) = new Person(firstName, lastName)
     | }
defined module Person

    接下来我们定义几个Person对象的实例,并测试distinct方法:

scala> val dale1 = new Person("Dale", "Cooper")
dale1: Person = Dale Cooper

scala> val dale2 = new Person("Dale", "Cooper")
dale2: Person = Dale Cooper

scala> val ed = new Person("Ed", "Hurley")
ed: Person = Ed Hurley

scala> val list = List(dale1, dale2, ed)
list: List[Person] = List(Dale Cooper, Dale Cooper, Ed Hurley)

scala> val uniques = list.distinct
uniques: List[Person] = List(Dale Cooper, Ed Hurley)


易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!