ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

Java实现 LeetCode 823 带因子的二叉树(DP)

2021-07-09 16:53:22  阅读:189  来源: 互联网

标签:10 Java DP int vi vj 二叉树 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);
    }
 
}
   

标签:10,Java,DP,int,vi,vj,二叉树,dp
来源: https://blog.51cto.com/u_14827074/3032179

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有