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

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 } }));
	}
}

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;
	}