最近在写实验的过程中,遇到在社交关系网络中,计算两个人物之间的最短距离的问题,恰巧之前也写过在无向图中,计算两个顶点之间的最短距离的问题,两者是一种问题,所以在这里记录一下。
关于人物之间的关系,这里我在Person类里设置一个Set集合的Rep来当做该人物的朋友圈(对于无向图来说,这个Set集合就用来存放与之相连的所有顶点了),见代码:
public class Person {
String name;
Set<Person> friends = new HashSet<Person>();
Person(String name) {
this.name = name;
}
/* 调取该人物的名字 */
public String getName() {
return this.name;
}
}
人物对象建立完毕,就要开始进入主题了。既然是人物A到人物B的最短距离,就要涉及到广度优先搜索了,以下是我的计算思路:
(1):将人物A的朋友(朋友圈里的所有人)添加到一个待查询的队列里(只添加没有查询过的人物);
(2):从前往后遍历队列,分别对每个人物的朋友圈进行遍历,遇到未添加到队列里的人物,则添加到队列里,直到队列里的所有人物都被访问过,结束;
(2.1):将每一层添加到队列里的人数记录到一维数组中;
(2.2):在查找到目标人物的同时,记录其在队列里的下标,结束遍历