Java实现 LeetCode 823 带因子的二叉树(DP)
2023-09-14 08:58:01 时间
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);
}
}
相关文章
- Java实现LeetCode 5448. 判断路径是否相交 (暴力)
- Java实现 LeetCode 732 我的日程安排表 III(暴力 || 二叉树)
- Java实现 LeetCode 654 最大二叉树(递归)
- Java实现 LeetCode 637 二叉树的层平均值(遍历树)
- Java实现 LeetCode 637 二叉树的层平均值(遍历树)
- Java实现 LeetCode 1162 地图分析(可以暴力或者动态规划的BFS)
- Java实现 LeetCode 563 二叉树的坡度(又是一个遍历树)
- Java实现 LeetCode 456 132模式
- Java实现 LeetCode 440 字典序的第K小数字
- Java实现 LeetCode 331 验证二叉树的前序序列化
- Java实现 LeetCode 268 缺失数字
- Java实现 LeetCode 222 完全二叉树的节点个数
- Java实现 LeetCode 165 比较版本号
- Java实现 LeetCode 145 二叉树的后序遍历
- Java实现 LeetCode 144 二叉树的前序遍历
- Java实现 LeetCode 144 二叉树的前序遍历
- Java实现 LeetCode 124 二叉树中的最大路径和
- Java实现 LeetCode 113 路径总和 II
- Java实现 LeetCode 107 二叉树的层次遍历 II(二)
- Java实现 LeetCode 103 二叉树的锯齿形层次遍历
- Java实现 LeetCode 103 二叉树的锯齿形层次遍历
- Java实现 LeetCode 107 二叉树的层次遍历 II(二)
- Java实现 LeetCode 106 从中序与后序遍历序列构造二叉树
- Java实现 LeetCode 51 N皇后
- Java实现LeetCode_0027_RemoveElement
- 【LeetCode-面试算法经典-Java实现】【144-Binary Tree Preorder Traversal(二叉树非递归前序遍历)】