給定一個整數 n,求以 1 ... n 爲節點組成的二叉搜索樹有多少種?
n = 3時,有5種
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
思路:
假設n個節點存在二叉排序樹的個數是G(n),令f(i)爲以i爲根的二叉搜索樹的個數,則:
G(n)=f(1)+f(2)+f(3)+f(4)+...+f(n)
當i爲根節點時,其左子樹節點個數爲i-1個,右子樹節點爲n-i,則:
f(i)=G(i−1)∗G(n−i)
兩個公式可以得到 卡特蘭數 公式:
G(n)=G(0)∗G(n−1)+G(1)∗(n−2)+...+G(n−1)∗G(0)
public class NumTrees {
public int numTrees(int n) {
int[] dp = new int[n + 1];
dp[0] = 1;
dp[1] = 1;
if(n <= 1) {
return 1;
}
//G(n)=G(0)∗G(n−1)+G(1)∗(n−2)+...+G(n−1)∗G(0)
for (int i = 2; i < n + 1; i++) {
for (int j = 0; j < i; j++) {
//從2開始遞推,遞推的方式空間複雜度:O(n)
dp[i] += dp[j] * dp[i - 1 - j];
}
}
return dp[n];
}
public static void main(String[] args) {
NumTrees numTrees = new NumTrees();
int i = numTrees.numTrees(5);
System.out.println(i);
}
}