LeetCode 95——不同的二叉搜索树 II

1. 题目

《LeetCode 95——不同的二叉搜索树 II》

2. 解答

《LeetCode 95——不同的二叉搜索树 II》 构建二叉搜索树,其中,任意数字都可以作为根节点来构建二叉搜索树。当我们将某一个数字作为根节点后,其左边数据将构建为左子树,右边数据将构建为右子树。因此,这是一个递归问题。

假设序列为 《LeetCode 95——不同的二叉搜索树 II》,若以第 《LeetCode 95——不同的二叉搜索树 II》 个数据为根节点,其左边数据 《LeetCode 95——不同的二叉搜索树 II》 可以构建出左子树 left_tree,右边数据 《LeetCode 95——不同的二叉搜索树 II》 可以构建出右子树 right_tree。它们都存储在一个向量中,因此,我们需要遍历左右子树所有可能的情况,分别构建二叉搜索树。

其中要特别注意,如果某一个子树向量为空,我们需要在向量中添加一个空指针,保证循环进行一次,这时候只需要遍历另一个子树即可。

/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */
class Solution {
public:
    vector<TreeNode*> generateTrees(int n) {
        
        return generateTrees(1, n);
    }
    
    vector<TeeNode*> generateTrees(int begin, int end) {
        
        vector<TreeNode*> all_tree;     
        vector<TreeNode*> left_tree;
        vector<TreeNode*> right_tree;
        TreeNode *tree = NULL;

        if (begin > end) return all_tree;

        for (int i = begin; i <= end; i++)
        {
            left_tree = generateTrees(begin, i - 1);
            right_tree = generateTrees(i + 1, end);
            
            // 左右子树如果为空的话需要循环一次
            if (left_tree.size() == 0) left_tree.push_back(NULL);
            if (right_tree.size() == 0) right_tree.push_back(NULL);
            for (int j = 0; j < left_tree.size(); j++)
            {         
                for (int k = 0; k < right_tree.size(); k++)
                {
                    tree = new TreeNode(i); // 每次都需要建一个新树
                    tree->left = left_tree[j];
                    tree->right = right_tree[k];
                    all_tree.push_back(tree);
                }
            }
        }
        
        return all_tree;
    }
};
复制代码

获取更多精彩,请关注「seniusen」!

《LeetCode 95——不同的二叉搜索树 II》

    原文作者:二叉树遍历
    原文地址: https://juejin.im/post/5bf4d4abf265da614f6fee7a
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞