不同的二叉搜索树 II（Leetcode-95）-递归分治&记忆化搜索

题目

• 原题连接：https://leetcode-cn.com/problems/unique-binary-search-trees-ii/
  

知识点

• 递归分治
• 记忆化搜索

思路

• 一开始做没有思路，想着怎么找状态去进行回溯，因为本身一棵树左右子树都是不固定的，不太好写
• 对于二叉搜索树：左子树的元素一定小于根节点，右子树的元素一定大于根节点，这为我们提供了递归分治的条件，我们在一开始设置范围去表示树左右节点的范围（int left, int right）。接着我们在该范围通过for循环去设置根节点。那么，对于根节点i来说，左子树的范围就是（left, i - 1），而右子树的范围就是（i + 1, right），这样我们递归生成树generate就可以获得所有可能的左子树（Line 22）与右子树（Line 23）必然是满足二叉搜索树条件的。（递归终止的条件就是left > right，在仅有一个叶子节点时生成左右子树时返回）。最后，两两匹配即是所有可能的二叉搜索树（Line 25-32）。
• 记忆化搜索：我们可以（int left, int right）记录下所有的状态（Line 34），对于已经生成过的情况直接返回即可（Line 18）。

代码

class Solution {
public:
    vector<TreeNode *> m[9][9];  // 范围不是很大，所以直接用数组记录状态了

    vector<TreeNode *> generate(int left, int right) {
        if (left > right) return {nullptr};
        if (!m[left][right].empty()) return m[left][right];

        vector<TreeNode *> trees;
        for (int i = left; i <= right; i++) {
            vector<TreeNode *> left_all = generate(left, i - 1);
            vector<TreeNode *> right_all = generate(i + 1, right);
			
			// 两两组合
            for (TreeNode *l:left_all) {
                for (TreeNode *r:right_all) {
                    TreeNode *root = new TreeNode(i);  // 当前选择的根节点
                    root->left = l;
                    root->right = r;
                    trees.emplace_back(root);
                }
            }
        }
        m[left][right] = trees;
        return trees;
    }

    vector<TreeNode *> generateTrees(int n) {
        return generate(1, n);
    }
};