JAVA程序设计：按公因数计算最大组件大小（LeetCode：952）

给定一个由不同正整数的组成的非空数组 A，考虑下面的图：

有 A.length 个节点，按从 A[0] 到 A[A.length - 1] 标记；
只有当 A[i] 和 A[j] 共用一个大于 1 的公因数时，A[i] 和 A[j] 之间才有一条边。
返回图中最大连通组件的大小。

 

示例 1：

输入：[4,6,15,35]
输出：4

示例 2：

输入：[20,50,9,63]
输出：2

示例 3：

输入：[2,3,6,7,4,12,21,39]
输出：8

 

提示：

1 <= A.length <= 20000
1 <= A[i] <= 100000

思路：并查集。我们将数组中的每个数和他们的因子关联起来，这样假如说两个数有公共因子的话他们俩一定是在一个集合中的，妥妥的并查集。

class Solution {

	private int[] f;

    public int largestComponentSize(int[] A) {
    	
    	int ans=0;
    	int n=A.length;
    	f=new int[100005];
    	int[] sum=new int[100005];
    	
    	for(int i=0;i<=100000;i++)
    		f[i]=i;
    	
    	for(int i=0;i<n;i++) {
    		int last=A[i];
    		for(int j=2;j<=(int)Math.sqrt(A[i]);j++) {
    			if(A[i]%j==0) {
    				merge(last,j);
    				last=j;
    				merge(last,A[i]/j);
    				last=A[i]/j;
    			}
    		}
    	}
    	
    	for(int i=0;i<n;i++) {
    		int index=find(A[i]);
    		sum[index]++;
    		ans=Math.max(ans, sum[index]);
    	}
    	
    	return ans;
    	
    }
    
    private void merge(int x,int y) {
    	int t1=find(x);
    	int t2=find(y);
    	if(t1!=t2)
    		f[t2]=t1;
    }
    
    private int find(int x) {
    	if(f[x]==x)
    		return x;
    	return f[x]=find(f[x]);
    }
}