题目
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
状态转移方程
思路
二叉树的情况是他的左子树情况 * 右子树情况。dp[i]表示i个节点的时候树可能的情况。
以j为中间节点的左右子树的情况。
for(int j = 1; j < i; j++)
dp[i] += dp[j-1] * dp[i-j]
Base
由状态转移方程可知
dp[0]=1
dp[1]=1
答案
class Solution {
public int numTrees(int n) {
int[] dp = new int[n+1];
dp[0] = 1;
dp[1] = 1;
for (int i = 2; i < n + 1; i++) {
for (int j = 1; j <= i; j++) {
dp[i] += dp[j-1]*dp[i-j];
}
}
return dp[n];
}
}