JAVA程序设计：带因子的二叉树（LeetCode：823）

给出一个含有不重复整数元素的数组，每个整数均大于 1。

我们用这些整数来构建二叉树，每个整数可以使用任意次数。

其中：每个非叶结点的值应等于它的两个子结点的值的乘积。

满足条件的二叉树一共有多少个？返回的结果应模除 10 ** 9 + 7。

 

示例 1:

输入: A = [2, 4]
输出: 3
解释: 我们可以得到这些二叉树: [2], [4], [4, 2, 2]
示例 2:

输入: A = [2, 4, 5, 10]
输出: 7
解释: 我们可以得到这些二叉树: [2], [4], [5], [10], [4, 2, 2], [10, 2, 5], [10, 5, 2].
 

提示:

1 <= A.length <= 1000.
2 <= A[i] <= 10 ^ 9.

思路：我们先将A数组从小到大排序，并将所有数放入集合，便于查找。然后从小到大枚举树的树根，对于每个树根，我们考虑其中一个因子，通过相除得到另一个因子，用集合查找判断另一个因子是否是A数组里的元素，若是，更新该树根的值。

class Solution {
    public int numFactoredBinaryTrees(int[] A) {
    	
    	int len=A.length;
    	long ans=0;
    	int mod=1000000007;
    	Set<Integer> st=new HashSet<>();
    	Map<Integer,Long> map=new HashMap<>();
    	
    	for(int i=0;i<len;i++) {
    		map.put(A[i], 1L);
    		st.add(A[i]);
    	}
    	
    	Arrays.parallelSort(A);
    		
    	for(int i=1;i<len;i++) 
    		for(int j=0;j<i;j++) {
    			int tmp=A[i]/A[j];
    			if(A[i]%A[j]!=0 || !st.contains(tmp))
    				continue;
    			map.put(A[i], (map.get(A[i])+map.get(tmp)*map.get(A[j])%mod)%mod);
    		}
    	
    	for(Integer x : map.keySet())
    		ans=(ans+map.get(x))%mod;
    	
    	return (int)ans;
    }
}