96. Unique Binary Search Trees
- Total Accepted: 95631
- Total Submissions: 246245
- Difficulty: Medium
Given n, how many structurally unique BST's (binary search trees) that store values 1...n?
For example,
Given n = 3, there are a total of 5 unique BST's.
1 3 3 2 1 \ / / / \ \ 3 2 1 1 3 2 / / \ \ 2 1 2 3
分析:对于任意整数i,有多少种结果呢?首先,它的根节点设为j,j可能取1,...,i中的任意
的数,现在任取一个值j作为其根节点,则它的左子数将由1,...,j-1这些值组成。它们组合的BST有多少种呢?
显然这是和原问题性质一样的子问题。这是想到动态规划。对于右子数,它是由j+1,...,i组成。其实他们组成
的二叉树的所有可能与1,...,i-j这些值组成的BST相同,即j+1,...,i依次减去j。因此,右子树的问题也是与
原问题性质相同的子问题。当根节点的值是j时,结果是左右子树的组合,即两者相乘。注意,当i=0时,
初始值为1.因为是乘。所以,对于i,将j=1,...,i的结果求和,既是解。
AC的代码:
class Solution { public: int numTrees(int n) { if(n<=1) return n; vector f(n+1); f[0]=1; f[1]=1; for(int i=1;i<=n;i++) { int sum=0; for(int j=1;j<=i;j++) { sum+=f[j-1]*f[i-j]; } f[i]=sum; } return f[n]; } };