Java實現 LeetCode 823 帶因子的二叉樹（DP）

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.

PS：
直接找能%的，並且餘數爲0，
從小到大找，有剪枝操作

class Solution {
    public int numFactoredBinaryTrees(int[] A) {
        int size = A.length;
        Arrays.sort(A);
        long[] dp = new long[size];
        long ans = 1;
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < size; i++) map.put(A[i], i);

        dp[0] = 1;
        for (int i = 1; i < size; i++) {
            int vi = A[i];
            long curres = 1;
            for (int j = 0; j < i; j++) {
                int vj = A[j];
                if (vj * vj > vi) break;
                Integer nj;
                if (vi % vj == 0 && (nj = map.get(vi/vj)) != null) {
                    curres += dp[j] * dp[nj] * (nj == j ? 1 : 2);
                    curres %= 1000000007;
                }
            }
            ans += (dp[i] = curres);
        }
        return (int)(ans % 1000000007);
    }
 
}