题目来源http://ac.jobdu.com/problem.php?pid=1526
题目1526:朋友圈
时间限制:1 秒
内存限制:128 兆
特殊判题:否
提交:411
解决:100
-
题目描述:
-
假如已知有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个朋友圈。
-
输入:
-
输入包含多个测试用例,每个测试用例的第一行包含两个正整数 n、m,1=<n,m<=100000。接下来有m行,每行分别输入两个人的编号f,t(1=<f,t<=n),表示f和t是好友。 当n为0时,输入结束,该用例不被处理。
-
输出:
-
对应每个测试用例,输出在这n个人里一共有多少个朋友圈。
-
样例输入:
-
5 3 1 2 2 3 4 5 3 3 1 2 1 3 2 3 0
-
样例输出:
-
2 1
-
来源:
- 小米2013年校园招聘笔试题
思路: 并查集路径压缩算法
import java.io.*;
import java.util.*;
public class Main {
public static int n,m;
public static int pattern[];
public static void main(String[] args) {
Scanner sc=new Scanner(new BufferedInputStream(System.in));
PrintWriter pw=new PrintWriter(new BufferedOutputStream(System.out),true);
while(sc.hasNext()){
n=sc.nextInt();
m=sc.nextInt();
pattern=new int[n+1];
for(int i=1;i<=n;i++){
pattern[i]=i;
}
for(int i=0;i<m;i++){
int a=sc.nextInt();
int b=sc.nextInt();
union(a,b);
}
int ans=0;
for(int i=1;i<=n;i++){
if(pattern[i]==i)
ans++;
}
pw.println(ans);
}
}
public static void union(int a,int b){
int aa=fun(a);
int bb=fun(b);
if(aa==bb)
return;
if(aa>bb)
pattern[bb]=aa;
else
pattern[aa]=bb;
}
public static int fun(int aa){
int p=pattern[aa];
if(p==aa)
return p;
else return fun(p);
}
}