您的位置:首页 > 其它

一共有n个人,给定m对好友关系,好友的好友也算一个朋友圈,问n个人中一共有几个朋友

2016-10-26 17:30 197 查看
题目:假如已知有n个人和m对好友关系(存于数字r)。如果两个人是直接或间接的好友(好友的好友的好友...),则认为他们属于同一个朋友圈,请写程序求出这n个人里一共有多少个朋友圈。假如:n = 5 , m = 3 , r = {{1 , 2} , {2 , 3} , {4 , 5}},表示有5个人,1和2是好友,2和3是好友,4和5是好友,则1、2、3属于一个朋友圈,4、5属于另一个朋友圈,结果为2个朋友圈。

package pac;

public class Demo1 {

/**
* @param args
* 并查集的应用
*/
private int [] set=new int[1000]; //存放上级,如set[5]=3表示5的上级是3

public int find(int x){ //查看掌门人
int r;
r=x;
while(set[r]!=r){ //如果掌门人不是自己,一直向上查找
r=set[r];
}
return r;
}

public void join(int x,int y){ //让两个人做朋友
int fx=find(x); //找到到x的掌门
int fy=find(y); //找到y的掌门
if(fx!=fy) { //如果不是同一个人
set[fx]=fy;
}
}

public int friends(int n,int m,int [][] r){
int count=0,i;
for(i=1;i<=n;i++) //初始化并查集
set[i]=i;
for(i=0;i<m;i++)
join(r[i][0],r[i][1]);
for(i=1;i<=n;i++)
if(set[i]==i) count++;
return count;
}
public static void main(String[] args) {
// TODO Auto-generated method stub
System.out.println(new Demo1().friends(5, 3,
new int[][] { { 1, 2 }, { 2, 3 }, { 4, 5 } }));
}
}



对find()进行路径优化,由上图左边的结构变成右边的结构

public int find1(int x){ //查看掌门人,并进行路径优化
int r;
r=x;
while(set[r]!=r){ //如果掌门人不是自己,一直向上查找
r=set[r];
}
int i=x,j;
while(i!=r){
j=set[i];
set[i]=r;
i=j;
}
return r;
}


具体查看http://www.jianshu.com/p/72ea54bb26ab
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐