LeetCode - 95.unique-binary-search-trees-ii

思路

遍历 1 到 n，列出所有以 i 为根节点的二叉搜索树。

因为BST的特性，如果 i 为根节点，那么左子树由 [1, i - 1] 组成，右子树由 [i + 1, n] 组成。

[start, end] 模式。

1. 当一个节点的start > end，说明这个节点是null。
2. 当 start == end 时，说明只有一个数字，只有一个节点。
3. 遍历 start 到 end。获得左子树的所有组合，获得右子树的所有组合，以 i 为 root，两两组合左右子树节点。

代码

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

    private List<TreeNode> getAns(int start, int end) {
        List<TreeNode> ans = new ArrayList<>();
        if (start > end){
            ans.add(null);
            return ans;
        }
        if (start == end){
            ans.add(new TreeNode(start));
            return ans;
        }
        for (int i = start; i <= end ; i++) {
            List<TreeNode> leftTrees = getAns(start, i - 1);
            List<TreeNode> rightTrees = getAns(i + 1, end);
            for (TreeNode leftTree : leftTrees) {
                for (TreeNode rightTree : rightTrees) {
                    TreeNode root = new TreeNode(i);
                    root.left = leftTree;
                    root.right = rightTree;
                    ans.add(root);
                }
            }
        }
        return ans;
    }