[leetcode] 95. 不同的二叉搜索树 II

95. 不同的二叉搜索树 II

96. 不同的二叉搜索树的扩展,现在要求把生成的算出来
这种建树问题一般来说都是用递归来解,这道题也不例外,划分左右子树,递归构造。

和我们之前dfs稍微有点不一样了,这次需要返回值了

class Solution {
    public List<TreeNode> generateTrees(int n) {
        if (n == 0) return new ArrayList<>();
        return dfs(1, n);
    }


    List<TreeNode> dfs(int start, int end) {
        List<TreeNode> subTree = new ArrayList<>();
        if (start > end) subTree.add(null);
        else {
            for (int i = start; i <= end; i++) {
                List<TreeNode> left = dfs(start, i - 1);
                List<TreeNode> right = dfs(i + 1, end);
                for (int j = 0; j < left.size(); j++) {
                    for (int k = 0; k < right.size(); k++) {
                        TreeNode node = new TreeNode(i);
                        node.left = left.get(j);
                        node.right = right.get(k);
                        subTree.add(node);
                    }
                }
            }
        }
        return subTree;
    }
}
posted @ 2018-08-08 21:47 ACBingo 阅读(...) 评论(...) 编辑 收藏