LeetCode0095-不同的二叉搜索树II

LeetCode0095-不同的二叉搜索树

题目：

给定一个整数 n，生成所有由 1 … n 为节点所组成的 二叉搜索树 。

示例：

输入：3
输出：
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]
解释：
以上的输出对应以下 5 种不同结构的二叉搜索树：

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

提示：

0 <= n <= 8

分析：

二叉搜索树的关键的性质是根节点的值大于左子树所有节点的值，小于右子树所有节点的值，且左子树和右子树也同样为二叉搜索树。因此在生成所有可行的二叉搜索树的时候，假设当前序列长度为n，如果我们枚举节点的值为i，那么根据二叉搜索树的性质可得，左子树的节点集合为[1…i-1],右子树的节点集合为[i+1,n]。而左子树和右子树的生成相较于原问题是一个长度缩小的子问题。可以使用递归的方法来解决这个题目。

定义generateTree(start, end)表示当前值的集合为[start, end]，返回序列[start, end]生成的所有可行的二叉搜索树。

考虑枚举[start, end]中的值i为当前二叉搜索树的根，那么递归划分为[start, i-1] 和[i+1, end] 俩部分，递归的调用这俩部分，获得可行的左子树和右子树，那么最后一步只要从可行的左子树集合中选一棵，再从可行的右子树中选一棵，拼接到根节点，即为其中一个解。

递归的入口为[1, n] ,出口为start > end 的时候，当前二叉搜索树为空，返回空节点。

代码：

/**
 * 0095-不同的二叉搜索树II
 * 给定一个整数 n，生成所有由 1 ... n 为节点所组成的 二叉搜索树 。
 * <p>
 * <p>
 * <p>
 * 示例：
 * <p>
 * 输入：3
 * 输出：
 * [
 * [1,null,3,2],
 * [3,2,null,1],
 * [3,1,null,null,2],
 * [2,1,3],
 * [1,null,2,null,3]
 * ]
 * 解释：
 * 以上的输出对应以下 5 种不同结构的二叉搜索树：
 * <p>
 * 1         3     3      2      1
 * \       /     /      / \      \
 * 3     2     1      1   3      2
 * /     /       \                 \
 * 2     1         2                 3
 * <p>
 * <p>
 * <p>
 * 提示：
 * <p>
 * 0 <= n <= 8
 */

import java.util.LinkedList;
import java.util.List;

/**
 * Definition for a binary tree node
 */

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode() {
    }

    TreeNode(int val) {
        this.val = val;
    }

    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}

/**
 * 递归实现
 */
class Solution {
    public List<TreeNode> generateTrees(int n) {
        if (n == 0) {
            return new LinkedList<TreeNode>();
        }
        return generateTrees(1, n);
    }

    /**
     * 重载方法
     *
     * @param start
     * @param end
     * @return
     */
    public List<TreeNode> generateTrees(int start, int end) {
        List<TreeNode> allTrees = new LinkedList<>();
        if (start > end) {
            allTrees.add(null);
            return allTrees;
        }
        // 枚举可行根节点
        for (int i = start; i <= end; i++) {
            // 获得所有可行的左子树集合
            List<TreeNode> leftTrees = generateTrees(start, i - 1);
            // 获得所有可行的右子树集合
            List<TreeNode> rightTrees = generateTrees(i + 1, end);

            // 从左子树集合中选一棵左子树，从右子树集合中选一棵右子树，拼接到根节点上
            for (TreeNode left : leftTrees) {
                for (TreeNode right : rightTrees) {
                    TreeNode currTree = new TreeNode(i, left, right);
                    allTrees.add(currTree);
                }
            }
        }
        return allTrees;
    }
}

/**
 * 测试
 */
public class Study0095 {
    public static void main(String[] args) {
        List<TreeNode> result = new Solution().generateTrees(3);
        for (TreeNode t : result) {
            preOrder(t);
            System.out.println();
        }
    }

    /**
     * 测试--输出前序遍历结果
     *
     * @param root
     */
    private static void preOrder(TreeNode root) {
        if (root == null) {
            return;
        }
        System.out.print(root.val + "\t");
        preOrder(root.left);
        preOrder(root.right);
    }
}

结果：