题目分析:
- 对于给定的数n,求由n个数构成的所有不同二分查找树的数目。
解题思路:
动态规划
二分查找树:左子树节点值小于root节点值,右子树节点值大于root节点值。
n个节点,每个节点都可以作为root节点,然后将小于root节点值的节点全部放入左子树,将大于root节点值的节点全部放入右子树,然后计算出左子树放置的种类,和右子树放置的种类,两个种类数相乘,即为以此root节点值为根的二分查找树的个数。总上分析,可以利用动态规划进行求解。
动态规划的初始值:
n == 0时,为空树,故dp[0] = 1;
n == 1时,只有1个根节点,故dp[0] = 1;
n == 2时,此时有两种情况,故dp[2] = dp[0] * dp[1] + dp[1] * dp[0];
n == 3时,此时有三种情况,故dp[3] = dp[0] * dp[2] + dp[1] * dp[1] + dp[2] * dp[0];
…
n == i时,此时dp[i] = sum(dp[0,…,k] * dp[k + 1,…i], 0 <= k <= i - 1;
实现程序
class Solution { public: int numTrees(int n) { vector<int> num; num.push_back(1); for (int i = 1; i <= n; i++) { num.push_back(0); // 初始化num[i]值 if (i < 3) num[i] = i; else { // 利用动态规划的递推公式求解num[i] for (int j = 1; j <= i; j++) num[i] += num[j - 1] * num[i - j]; } } // 得到二分查找树的种类数 return num[n]; } };