码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode[96]-Unique Binary Search Trees

时间:2015-06-13 09:58:12      阅读:104      评论:0      收藏:0      [点我收藏+]

标签:binary   dp   动态规划   二叉树计数   

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

思路: 动态规划解题,dp[n]表示n个节点可以有dp[n]种不同的树,不管n为多少,先固定一个节点,剩余n-1个节点,分配给左右字数,然后把左子树个数乘以右子树的个数,

  • 初始值dp[0] = 1,dp[1] = 1,
  • dp[n] = dp[0] * dp[n-1] + dp[1] * dp[n-2] + …+ dp[i] * dp[n-1-i] +… + dp[n-1] * dp[0],也就是左边i个节点,右边n-1-i个节点。代码如下:

Code(c++):

class Solution {
public:
    int numTrees(int n) {
        vector<int> dp;
        dp.resize(n+1);//set  the length of vector to n+1
        for(int i = 0; i <= n; i++) {
            //dp[0] = 1 , dp[1] =1
            if(i<2){
                dp[i] = 1;
                continue;
            }
            //dp[n] = dp[0]*dp[n-1]+dp[1]*dp[n-2]+...+dp[n-1]*dp[0]
            for(int j = 1; j<=i; j++) {
                dp[i] += dp[j-1]*dp[i-j];
            }
        }
        return dp[n];
    }
};

Leetcode[96]-Unique Binary Search Trees

标签:binary   dp   动态规划   二叉树计数   

原文地址:http://blog.csdn.net/dream_angel_z/article/details/46475787

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!