题解 | #不同的二叉搜索树(一)#

不同的二叉搜索树(一)

https://www.nowcoder.com/practice/6e5df579ca694d4190304da6406c4dfe

动态规划 dp[i]表示i个节点对应的二叉搜索树个数
考虑边界的情况,即0个节点和1个节点的情况,那么构建的二叉搜索树只能是空树或只有根节点的树
转移方程:当前节点为i时,那么从[1,i]中的每个数都可以作为根节点,假设当前取j∈[1,i]为根节点,由于是构建二叉搜索树,那么左子树的节点只能选[1,j-1]范围的数,右子树只能选取[j+1,i]范围内的数。构建的时候只有左右子树节点个数有关
构建搜索二叉树的时候,根节点为j的所有可能的情况为 左子树的情况数(dp[i-1])✖右子树的情况数(dp[i-(j+1)+1]  ---> dp[i-j])。而节点个数为i的情况下所能构建的二叉搜索树的情况总数为每个节点作为根节点的情况数之和。
public class Solution {
    public int BSTCount (int n) {
        // write code here
        if(n < 2)
            return n;
        int[] dp = new int[n+1];
        dp[0] = 1;
        dp[1] = 1;
        
        for(int i=2;i<=n;i++){
            int ans = 0;
            // 枚举每个节点作为根节点
            for(int j=1;j<=i;j++){
                ans += dp[j-1]*dp[i-j];
            }
            dp[i] = ans;
        }
        return dp[n];
    }
}


全部评论

相关推荐

头像
不愿透露姓名的神秘牛友
05-24 12:16
点赞 评论 收藏
转发
头像
05-27 20:32
已编辑
深度学习
工行数据中心 偏运维养老 到手可能18w
点赞 评论 收藏
转发
点赞 收藏 评论
分享
牛客网
牛客企业服务