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.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-trees-with-factors
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

解题思路

class Solution {
public:
    int numFactoredBinaryTrees(vector<int>& A) {
        int ans = 1,len = A.size();
        int mod = 1e9+7;
        vector<long long> dp(len,1);
        sort(A.begin(),A.end());
        for(int i=1;i<len;++i){
            int tmp = A[i],j = i-1;
            while(j>=0 && A[j]>=sqrt(tmp)){
                if(tmp%A[j] == 0){
                    int l = 0,r = j,fid = tmp/A[j];
                    int midx,midv;
                    while(l <= r){
                        midx = (l+r)/2;
                        midv = A[midx];
                        if(midv == fid) break;
                        else if(midv > fid) r = midx-1;
                        else l = midx+1;
                    }
                    if(l<=r){
                        if(A[j] == midv) dp[i] = (dp[i] + dp[j]*dp[midx])%mod;
                        else dp[i] = (dp[i] + dp[j]*dp[midx]*2)%mod;
                    }
                }
                j--;
            }
            ans = (ans + dp[i])%mod;
        }
        return ans;
    }
};