LeetCode:96 不同的二叉搜索樹

給定一個整數 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);
    }
}

 

發佈了67 篇原創文章 · 獲贊 10 · 訪問量 6765
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章