LeetCode每日一题:823. 带因子的二叉树(2023.8.29 C++)

目录

823. 带因子的二叉树

题目描述:

实现代码与解析:

dp + hash

原理思路:


823. 带因子的二叉树

题目描述:

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

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

满足条件的二叉树一共有多少个?答案可能很大,返回 对 109 + 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] <= 109
  • arr 中的所有值 互不相同

实现代码与解析:

dp + hash

class Solution {
public:
    int mod = 1e9 + 7;
    
    int numFactoredBinaryTrees(vector& arr) {
        
        unordered_map f; // 以 first 为根的种类为 second
        sort(arr.begin(), arr.end());

        for (auto t: arr) f[t] = 1;

        for (int i = 0; i < arr.size(); i++)
        {
            for (int j = 0; j < i; j++)
            {
                long long x = arr[i], y = arr[j];
                if (x % y == 0 && f.count(x / y)) f[x] += f[y] * f[x / y] % mod;
            }
        }

        long res = 0;
        for (auto [a, b]: f)
            res = (res + b) % mod;

        return res;
    }
};

原理思路:

        dp含义:以 first 为 根 的种类为 second。

        这里为什么没用数组来存dp呢,如果我们这里用数组存,后面要判断一个数的因数是否存在arr中也要再定义一个hash,等于说这个map其两个作用,一个是dp,一个是hash。当然也可以不用hash,用双指针或者二分也可以找到。  

        遍历顺序肯定是从树底到顶,也就是从小到大,排序后开始遍历。

        若x 可整除 y 并且 两个因数y , x / y都在arr中,那么f[ i ] += f[x] += f[y] * f[x / y]; (种类是子树种类乘积)

        最后sum一下即可。

你可能感兴趣的:(LeetCode,leetcode,算法)