LeetCode第96题

96.不同的二叉搜索树

---

给定一个整数n，求以1…n为节点组成的二叉搜索树有多少种？

示例

输入：3

输出：5

给定n = 3，一共有5种不同结构的二叉搜索树

解题思路

这道题其实是一道动态规划题。假设 dp(n)为n个数所组成的二叉搜索树的数量，f(i)为以i为节点能构成几个二叉搜索树。那么dp(n) = f(1) + f(2) + f(3) + ... + f(n)

那么接下来我们看f(i)等于什么，我们都知道，二叉搜索树是任意节点的左孩子都小于该节点，任意节点的右孩子都大于该节点。换句话说，假设i为根节点的话，那么i左边有1,2,3,...,i-1这些节点（共i-1个），右边有i+1,i+2,...,n（共n-i个），那么也就是说f(i) = dp[i-1] * dp[n - i]，知道了f(i)也就可以求出dp[n]了。接下来让我们来进行代码实现吧！

代码实现

public class LeetCode_96 {
    public int numTrees(int n) {
        if (n == 0 || n == 1) {
            return 1;
        }
        int[] dp = new int[n + 1];
        dp[0] = 1;
        dp[1] = 1;
        for (int i = 2; i <= n; i++) {
            for (int j = 1; j <= i; j++) {
                dp[i] += dp[j - 1] * dp[i - j];
            }
        }
        return dp[n];
    }
    @Test
    public void test() {
        int result = numTrees(3);
        System.out.println(result);
    }
}

后记

动态规划问题找出递推式是关键。