本文主要是介绍从两两好友关系的记录中,找出三人中任意两人互为好友(三角好友)的Scala实现,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
曾经遇到一个问题:
有一个文件,其中每一行都记录了存在好友关系的两个人的userId。现要分析此文件,从两两朋友关系的记录中,找出所有的:三人中任意两人互为好友,即三角好友的存在,并打印出来
当时现场没有写出来。最近尝试编写了下,抛去读取文件等基础工作,先实现了三角好友关系查找的实现,编写代码如下:
/*** Created by leisu on 2018/7/19.*/
object TriangleFriendsRelationShip {def main(args: Array[String]) {//初始化一批数据val A = "A"val B = "B"val C = "C"val D = "D"val E = "E"val F = "F"val origin = List(A, B) :: List(A, C) :: List(A, D) :: List(B, C) :: List(B, D) :: List(C, D) :: List(C, E) :: List(C, F) :: List(E, F) :: Nilval oriSet: List[Set[String]] = origin.map(_.toSet)//调用方法处理val matchTupelList: List[Tuple2[Set[String], Set[String]]] = findCommonPoint(oriSet)val resultFinal = findMatchTriangle(oriSet, matchTupelList)//打印resultFinal.map(println)}/*** 找出List记录内,任两组数据中有共同顶点的匹配关系,组成新的Tuple 类似Set(A,B)+ Set(A,C) => Tuple2(Set(B,C),Set(A)) 这种结构* 并最终拼装成新list返回**/def findCommonPoint(list: List[Set[String]]): List[Tuple2[Set[String], Set[String]]] = list match {case headSet :: tail => {if (tail == Nil) Nilelse {val lMatch = for (entrySet <- tail;item = ((headSet | entrySet) -- (headSet & entrySet), (headSet & entrySet)) //找出合并差if item._1.size == 2 //只有合并差为2时(即有共同定点时),才筛选出来) yield itemlMatch ::: findCommonPoint(tail) //递归,此次处理结果与前面处理结果进行合并}}}/*** 根据第三条边,找出完整的三角匹配关系*/def findMatchTriangle(oriList: List[Set[String]], matchTupelList: List[Tuple2[Set[String], Set[String]]]): List[Set[String]] = {val resultList = for (entrySet <- oriList;matchTuple <- matchTupelList;result = entrySet | matchTuple._2if (entrySet == matchTuple._1)) yield resultresultList.distinct //去重}
}
代码中案例的执行i结果如下:
Set(A, B, C)
Set(A, B, D)
Set(A, C, D)
Set(B, C, D)
Set(C, E, F)Process finished with exit code 0
根据测试,此结果正确,完成功能
这篇关于从两两好友关系的记录中,找出三人中任意两人互为好友(三角好友)的Scala实现的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!