Leetcode - Unique Binary Search Trees II

Paste_Image.png

My code:

import java.util.ArrayList;
import java.util.List;

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<TreeNode> generateTrees(int n) {
        if (n < 0)
            return null;
        return generateTrees(1, n);
    }
    
    private List<TreeNode> generateTrees(int begin, int end) {
        ArrayList<TreeNode> tree = new ArrayList<TreeNode>();
        if (begin > end) {
            tree.add(null);
            return tree;
        }
        
        for (int i = begin; i <= end; i++) {
            List<TreeNode> left = generateTrees(begin, i - 1);
            List<TreeNode> right = generateTrees(i + 1, end);
            for (TreeNode leftNode : left) {
                for (TreeNode rightNode : right) {
                    TreeNode root = new TreeNode(i);
                    root.left = leftNode;
                    root.right = rightNode;
                    tree.add(root);
                }
            }
        }
        return tree;
    }
}

My test result:


Paste_Image.png

这道题目也没做出来。但是看了解题思路后,觉得太强大了。这么短的十几行代码,其实做了超多事情,而且这些事情,在我看来是很难想象的。
比如,假定给定n,那么,他就是构造了, uniqueBST1(n) 棵树。
然后,把这些树的头结点,全部插入了list。
generateTrees 会返回一个arraylist,这是,[begin, end] 范围内,所有可构造出来的树的头结点的集合。是会有重复值的头结点的。
比如,头结点val = 5的头结点可能会有好多个,但是他们是不同的,因为他们分别代表了不同结构的树。
然后, [begin, i - 1] and [i + 1, end] 会返回左右两棵子树的头结点集合。
假设,分别为 ,n个 m个。
那么,这里就会有 n * m 中结构,共同点是,头结点都是 i
然后把这 n * m 个 值为val的头结点,插入list中,返回给上级。
直到最后,
返回的,都是,根节点,而且,每个根节点下面,都连着一个截然不同的树。
然后这么复杂的事,就在这十几行代码里面实现了。
递归,DP,都是用简洁的思想去实现复杂的事,是真的特别特别复杂,人力无法及的事。

**
总结: Tree, DP, recursion, permutation
**

Anyway, Good luck, Richardo!

My code:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<TreeNode> generateTrees(int n) {
        List<TreeNode> ret = new ArrayList<TreeNode>();
        if (n <= 0) {
            return ret;
        }
        
        return helper(0, n - 1);
    }
    
    private List<TreeNode> helper(int begin, int end) {
        List<TreeNode> ret = new ArrayList<TreeNode>();
        if (begin > end) {
            return ret;
        }
        else if (begin == end) {
            ret.add(new TreeNode(begin + 1));
            return ret;
        }
        
        for (int i = begin; i <= end; i++) {
            List<TreeNode> leftTrees = helper(begin, i - 1);
            List<TreeNode> rightTrees = helper(i + 1, end);
            if (leftTrees.size() == 0) {
                for (TreeNode right : rightTrees) {
                    TreeNode root = new TreeNode(i + 1);
                    root.right = right;
                    ret.add(root);
                }
            }
            else if (rightTrees.size() == 0) {
                for (TreeNode left : leftTrees) {
                    TreeNode root = new TreeNode(i + 1);
                    root.left = left;
                    ret.add(root);
                }
            }
            else {
                for (TreeNode left : leftTrees) {
                    for (TreeNode right : rightTrees) {
                        TreeNode root = new TreeNode(i + 1);
                        root.left = left;
                        root.right = right;
                        ret.add(root);
                    }
                }
            }
        }
        
        return ret;
    }
}

几遍就过了。
思路还是和第一题差不多,只不过这次不能用cache了。
思路还是 burst balloon 的思路。分割,画块,处理。
似乎也没有体现出DP的思想。可能就是自顶向下吧。

Anyway, Good luck, Richardo! -- 08/25/2016

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容