LeetCode 96. Unique Binary Search Trees

96. Unique Binary Search Trees

题目

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

示例:

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

  1         3     3      2      1
   \       /     /      / \      \
    3     2     1      1   3      2
   /     /       \                 \
  2     1         2                 3

解决方案

要求节点个数为n时二叉搜索树的可能个数
n=0  二叉搜索树个数为1
n=1  二叉搜索树个数为1
n=2   左子树节点个数为0且右子树节点个数为1或者左子树节点个数为1且右子树节点个数为0这两种情形即N(n)=N(0)*N(1)+N(1)*N(0)
可推导得求二叉搜索树个数公式为
N(n)=N(0)*N(n-1)+N(1)*N(n-2)+...+N(i)*N(n-1-i)+...+N(n-1)*N(0)

class Solution {
public:
    int numTrees(int n) {
      if(n<2)
         return 1;
      int list[n+1];
      for(int i=0;i<n+1;++i)
          list[i]=0; 
      list[0] = 1;
      list[1] = 1;
      for(int i=2;i<n+1;++i)
          for(int j=0;j<i;++j)
              list[i] +=list[j] * list[i-1-j];
      return list[n];
    }
};