LeetCode每日一题 823. 带因子的二叉树

题干：

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

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

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

示例 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].

提示：

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

对于任何一个非叶节点的值，它必然是它的两个子节点的值的乘积。因此，我们可以考虑将某个整数作为一个非叶节点，然后从数组中选择合适的整数作为它的左右子节点，使得满足题目条件。

我们可以使用动态规划来解决这个问题。我们定义一个动态规划数组 dp，其中 dp[i] 表示使用数组的前 i 个整数构建满足条件的二叉树的数量。初始时，每个整数本身可以作为一个只有根节点的二叉树，因此 dp[i] 的初始值为 1。

接下来，我们遍历数组中的每个整数 arr[i]，并尝试将其作为一个非叶节点。我们可以遍历之前的整数 arr[j]（其中 j 取值范围为 0 到 i-1），将 arr[j] 作为左子节点的值，而右子节点的值可以从 arr[i] / arr[j] 中选择。如果 arr[i] / arr[j] 也存在于数组中，那么说明我们可以使用 arr[i] 作为非叶节点，将 arr[j] 作为左子节点，而 arr[i] / arr[j] 作为右子节点，构建一棵满足条件的二叉树。此时，可以将 dp[i] 更新为 dp[i] + dp[j] * dp[arr[i] / arr[j]]，表示在这种情况下新增的二叉树数量。

最后，遍历整个 dp 数组，将所有值累加起来，即得到了使用整数数组 arr 构建满足条件的二叉树的总数量。

为了避免计算过程中的溢出问题，我们可以使用取余操作。最终的答案应该是 dp[n] % (10^9 + 7)，其中 n 是数组 arr 的长度。

将上述思路转化为代码，我们可以得到以下 C++ 实现：

#include <iostream>
#include <vector>
#include <unordered_map>

const int MOD = 1e9 + 7;

class Solution {
public:
    int numFactoredBinaryTrees(vector<int>& arr) {
        sort(arr.begin(), arr.end());

        int n = arr.size();
        vector<long long> dp(n, 1);  // 使用 long long 防止溢出

        unordered_map<int, int> indexMap;
        for (int i = 0; i < n; ++i) {
            indexMap[arr[i]] = i;
        }

        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < i; ++j) {
                if (arr[i] % arr[j] == 0) {
                    int k = arr[i] / arr[j];
                    if (indexMap.find(k) != indexMap.end()) {
                        dp[i] = (dp[i] + dp[j] * dp[indexMap[k]]) % MOD;
                    }
                }
            }
        }

        long long totalTrees = 0;
        for (long long count : dp) {
            totalTrees = (totalTrees + count) % MOD;
        }

        return static_cast<int>(totalTrees);
    }
};

int main() {
    Solution solution;
    vector<int> arr = {2, 4, 5, 10};
    cout << solution.numFactoredBinaryTrees(arr) << endl;
    return 0;
}