95. Unique Binary Search Trees II

Given an integer n, generate all structurally unique BST's (binary search trees) that store values 1...n.
For example,
Given n = 3, your program should return all 5 unique BST's shown below.

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

解法类似 241题:http://www.jianshu.com/p/91b3c4352b43

Solution1:Divide and Conquer

以每个num=0...n都尝试作为起点,以num分成左右两部分part1 for [0..num - 1], part2 for [num + 1, n] (Divide),recursiely计算part1和part2的建树结果,再将这两个组合结果Conquer构成从当前num分的结果。
Time Complexity: ? Space Complexity: ?

Solution2:Divide and Conquer + Hashmap

Solution1中含有重复计算,2在Solution1的基础上 将已求过的[start, end]的结果 Map<String, List<TreeNode>> cached_tree保存下来,缓存避免重复计算。
Time Complexity: ? Space Complexity: ?

Solution3:DP写法

ret[n] 保存到n的结果(所有的unique BST)
根据96. Unique Binary Search Trees: http://www.jianshu.com/p/a49119ef97c3
的DP递推表达式

G(n): the number of unique BST for a sequence of length n.
F(i, n), 1 <= i <= n: the number of unique BST, where the number i is the root of BST, and the sequence ranges from 1 to n.

G(0)=1, G(1)=1. 
G(n) = F(1, n) + F(2, n) + ... + F(n, n). 
F(i, n) = G(i-1) * G(n-i)   1 <= i <= n 
So, 
=>    G(n) = G(0) * G(n-1) + G(1) * G(n-2) + … + G(n-1) * G(0) 

思路相同,这不过这里是把G当作实际的BSTs,G(0) 和 G(n-1) 做组合,G(1) * G(n-2)做组合,...都加到结果中。做组合时树:node.value = j, 左侧的是left children,右侧的是right children(需shift),
Example G(4) 如下图所示:

屏幕快照 2017-09-04 下午11.43.50.png

n = sum of the total number of unique binary trees
Time Complexity: n? Space Complexity: n?

Solution1 Code:

class Solution1 {
    public List<TreeNode> generateTrees(int n) {
        if(n == 0) return new ArrayList<TreeNode>();
        return genTrees(1,n);
    }
        
    public List<TreeNode> genTrees (int start, int end) {
        List<TreeNode> list = new ArrayList<TreeNode>();
        if(start > end) list.add(null);
        else if(start == end) list.add(new TreeNode(start));
        else {
            // start < end
            List<TreeNode> left,right;
            for(int i = start; i <= end; i++) {
                // Divide
                left = genTrees(start, i - 1);
                right = genTrees(i + 1, end);

                // Conquer
                for(TreeNode lnode: left) {
                    for(TreeNode rnode: right) {
                        TreeNode root = new TreeNode(i);
                        root.left = lnode;
                        root.right = rnode;
                        list.add(root);
                    }
                }
            }
        }
        return list;
    }
}

Solution2 Code:

class Solution2 {
    public List<TreeNode> generateTrees(int n) {
        if(n == 0) return new ArrayList<TreeNode>();
        return genTrees(1,n);
    }
    
    private Map<String, List<TreeNode>> cached_tree = new HashMap<>();
        
    public List<TreeNode> genTrees (int start, int end) {
        // find if there has been cached already
        String key = "" + start + "." + end;
        if(cached_tree.containsKey(key)) return cached_tree.get(key);
        
        List<TreeNode> list = new ArrayList<TreeNode>();
        if(start > end) list.add(null);
        else if(start == end) list.add(new TreeNode(start));
        else {
            // start < end
            List<TreeNode> left,right;
            for(int i = start; i <= end; i++) {
                // Divide
                left = genTrees(start, i - 1);
                right = genTrees(i + 1, end);

                // Conquer
                for(TreeNode lnode: left) {
                    for(TreeNode rnode: right) {
                        TreeNode root = new TreeNode(i);
                        root.left = lnode;
                        root.right = rnode;
                        list.add(root);
                    }
                }
            }
        }
        cached_tree.put(key, list);
        return list;
    }
}

Solution3 Code:

class Solution3 {
    // G(0)=1, G(1)=1. 
    // G(n) = F(1, n) + F(2, n) + ... + F(n, n). 
    // F(i, n) = G(i-1) * G(n-i)   1 <= i <= n 
    // =>
    // G(n) = G(0) * G(n-1) + G(1) * G(n-2) + … + G(n-1) * G(0) 
    
    public List<TreeNode> generateTrees(int n) {
        if(n == 0) return new ArrayList<TreeNode>();
        
        // dp init
        List<TreeNode>[] res = new List[n+1];
        res[0] = new ArrayList();
        res[0].add(null);
        res[1] = new ArrayList();
        res[1].add(new TreeNode(1));
        
        // dp
        for(int i = 2; i <= n; i++){
            res[i] = new ArrayList();
            for(int j = 1; j <= i; j++){
                for(TreeNode nodeL: res[j - 1]){
                    for(TreeNode nodeR: res[i - j]){
                        TreeNode node = new TreeNode(j);
                        node.left = nodeL;
                        node.right = clone(nodeR, j);
                        res[i].add(node);
                    }
                }
            }
        }
        return res[n];
    }
    
    private TreeNode clone(TreeNode n, int offset) {
        if (n == null) {
            return null;
        }
        TreeNode node = new TreeNode(n.val + offset);
        node.left = clone(n.left, offset);
        node.right = clone(n.right, offset);
        return node;
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,425评论 4 361
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,058评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,186评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,848评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,249评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,554评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,830评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,536评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,239评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,505评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,004评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,346评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,999评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,060评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,821评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,574评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,480评论 2 267

推荐阅读更多精彩内容