美文网首页动态规划
Leetcode-96:不同的二叉搜索树

Leetcode-96:不同的二叉搜索树

作者: 小北觅 | 来源:发表于2018-11-22 16:37 被阅读11次

    描述:
    给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?
    示例:

    输入: 3
    输出: 5
    解释:
    给定 n = 3, 一共有 5 种不同结构的二叉搜索树:

    思路:

    二叉搜索树:
    ①若任意节点的左子树不为空,则左子树上所有结点的值均小于它根节点的值
    ②若任意节点的右子树不为空,则右子树上所有结点的值均大于它根节点的值
    ③任意结点的左、右子树也是二叉搜索树

    在本题中:
    设dp(n)表示有n个结点时搜索二叉树有多少种可能,则

    Ⅰ.当头结点的值为1时,左子树为空,右子树上有n - 1个结点,右子树的搜索二叉树个数为dp(n - 1)

    Ⅱ.当头结点的值为i(1 < i < n)时,左子树由结点1—>i-1构成,右子树由结点i+1—>n构成;左子树的搜索二叉树个数为dp(i-1),右子树的搜索二叉树个数为dp(n - i);此时搜索二叉树总的个数为dp(i - 1) * dp(n - i)

    Ⅲ.当头结点的值为n时,右子树为空,左子树上有n - 1个结点,左子树的搜索二叉树个数为dp(n - 1)

    所以,n个结点时搜索二叉树的个数上述三个步骤的和。

    递推公式:
    dp(n)=dp(0)dp(n-1)+dp(1)dp(n-2)+dp(2)dp(n-3)+…+dp(n-1)dp(0)

    class Solution {
        public int numTrees(int n) {
            //dp[i]表示有i个结点时二叉树有多少种可能
            int[] dp = new int[n + 1];
            //初始化
            dp[0] = 1;
            dp[1] = 1;
            
            //因为计算dp[n]需要知道dp[0]--->dp[n-1]。所以第一层循环是为了求dp[i]
            for (int i = 2; i <= n; i++) {
                //当有i个结点时,左子树的节点个数可以为0-->i-1个。剩下的是右子树。
                for (int j = 0; j < i; j++) {
                    dp[i] += dp[j] * dp[i - j - 1];
                }
            }
            return dp[n];
        }
    }
    

    相关文章

      网友评论

        本文标题:Leetcode-96:不同的二叉搜索树

        本文链接:https://www.haomeiwen.com/subject/feikqqtx.html