leetcode 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.
  

  来源：力扣（LeetCode）
  题目

分析

• 首先这道题目的元素是可以复用的,并且可以摞起来,我们可以按照以不同的根作为节点的形成的树的个数加起来就是答案,这道题有明显的子结构,就是小的根可以作为大元素的子节点再次利用
  就比如[2,4,8]数据,我们我们在计算了4可以形成两种以4为根节点的树后,在计算以8为根的节点的时候,就看到之前记录的4,我们直接拿过来用,因为可以复用,所以我们直接到数组里找有没有(8/4=)2,有就可以左右子树不同的组合就是为根节点的树的个数的乘积

  tmp=(tmp+(dp[j]*dp[pos])%mod);
  

  所以转移方程就是这个,动态规划的思路就基本出来了,剩下的就是细节.

  • 二分查找使用的是简单的查数,使用左右闭区间,正好简单复习一下要点,并且为了快1点,所以右边界使用传参的i,减小查找范围.

总体

• 复杂度为o(n^2*log(n))
  首先使用排序(他没说给的是有序的,虽然示例是有序的,实际测试用例也不是有序的),方便我们进行动态规划遍历,以及后面的二分查找.之后从最小元素开始遍历i,计算以A[i]为根的树的个数,首先肯定有一个有且只有自己的树,相当于动规中的base.tmp是计算i的树个数的变量,所以先初始化为1,之后遍历i之前的元素,看是否能够除尽,能够除尽就去数组里找有没有除尽的这个数,使用二分查找,返回索引,之后使用转移方程计算累加

尚待优化

1. 首先由于取模的时候,两个int相乘会超,leetcode会报错,这个我想处理,但是忘了怎 么搞了,所以为了偷懒使用了vector直接用longlong来存,幸好时间还可以.也希望有人教教我怎么搞这个
   错误提示
   tmp=(tmp+(dp[j]*dp[pos])%mod); //26.
   Line 26: Char 40: runtime error: signed integer overflow: 2271 * 1161903 cannot be represented in type ‘int’ (solution…)
   …会了,改成

tmp=(tmp+((