C++实现带因子的二叉树

题目

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

用这些整数来构建二叉树，每个整数可以使用任意次数。其中：每个非叶结点的值应等于它的两个子结点的值的乘积。

满足条件的二叉树一共有多少个？答案可能很大，返回 对 109 + 7 取余 的结果。

限制条件：

1 <= arr.length <= 1000
2 <= arr[i] <= 109
arr 中的所有值 互不相同

示例

示例 1:

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

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

思路

由于要求数组arr中元素构成的二叉树的每个非叶结点的值应等于它的两个子结点的值的乘积，若对于某个root节点，存在left子节点和right子节点均属于数组元素，且知道left子节点所构成的二叉树的个数m，以及right子节点所构成的二叉树的个数n，则m*n就表示以left和right元素为子节点，且父节点为root的二叉树的个数。

代码

对数组arr进行从小到大的排序，每次判断当前元素能否被更小的数组元素的乘积表示，若能就按照上述思路计算可能的二叉树个数，从最小元素开始计算能保证得到以该元素为root的所有可能二叉树的个数。初始条件是数组中的每个元素都能单独组成一颗二叉树，采用std::map的C++ 代码实现如下：

class Solution {
public:
    int numFactoredBinaryTrees(vector<int>& arr) {
        // long long避免越界
        long long res = 0, n = arr.size(); 
        int mod = 1e9+7;
        // 从小到大顺序的map，存储以该key为root的树的个数
        map<int, long long> hashmap; 
        for(int i = 0; i < n; i++){
            hashmap[arr[i]] = 1; // 每个元素单独为一个树
        }
        // 对排好序的map进行遍历，判断当前key能否表示成之前key的积
        for(map<int, long long>::iterator it = hashmap.begin(); it !=hashmap.end(); it++){
            for(map<int, long long>::iterator pre = hashmap.begin(); pre !=it; pre++){
                // 不能表示为数组元素的乘积形式
                if (it->first % pre->first != 0) continue;
                int left = it->first / pre->first;
                // 数组中不存在该元素继续遍历
                if (!hashmap[left]) continue;
                // 计算能表示的二叉树的个数
                // 第二重循环可以通过判断pre->first <= sqrt(it->fisrt)进行剪枝
                it->second += hashmap[left] * pre->second % mod;
            }
            // 从小到大的顺序保证了当前元素为root节点的所有二叉树的个数
            res = (res + it->second) % mod;
        }
        return res;
    }
};

复杂度分析

空间复杂度主要为std::map的空间复杂度$O(n)$，时间复杂度为双重循环$O(n^2)$。