leetcode 823

DP，先把arr排序，dp[i]表示以arr[i]为根节点的树有多少颗。然后两层遍历，如果arr[i] / arr[j] (j < i) 存在，则更新dp[i]的值。这里我一开始是找arr[i] * arr[j]是否在arr中存在，但是显然有可能超出int范围。如果用python的话就不太需要注意类型范围上限的问题。

class Solution {
    public int numFactoredBinaryTrees(int[] arr) {
        int MOD = (int)(Math.pow(10, 9)) + 7;
        Arrays.sort(arr);
        long[] dp = new long[arr.length];
        Map<Integer, Integer> position = new HashMap<>();
        for(int i = 0; i < dp.length; i++){
            position.put(arr[i], i);
            dp[i] = 1;
        }
        for(int i = 1; i < arr.length; i++){
            for(int j = 0; j < i; j++){
                if(arr[i] % arr[j] == 0 && position.containsKey(arr[i] / arr[j])){
                    int k = position.get(arr[i] / arr[j]);
                    dp[i] = (dp[i] + dp[k] * dp[j]) % MOD;
                }
            }
        }
        
        long sum = 0;
        for(long d : dp){
            sum += d % MOD;
        }
        return (int)(sum % MOD);
    }
}