Leetcode 95. Unique Binary Search Trees II

题目:

Given an integer n, generate all structurally unique BST's (binary search trees) that store values 1 ... n.

Example:

Input: 3
Output:
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]
Explanation:
The above output corresponds to the 5 unique BST's shown below:

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

需要注意的是,这里的Output表示的是对应结果树,返回类型是vector<TreeNode*> result ,即每个结果树的根节点。

解题思路:

要生成搜索二叉树,容易想到每一个节点i,i∈[1,n],其左子树的每个节点应在[1,i-1]范围内,而其右子树的每个节点应在[i+1,n]范围内,若该范围不合理,则其对应子树为NULL。因此,可以考虑递归的方式,每次选取一个结点为根,然后根据左右子树范围有限的条件递归求解其左右子树的所有结果,再根据左右子树的返回子树,将所有结果连接到根结点上。

代码如下:

class Solution {
public:
    vector<TreeNode*> generate(int start,int end) {
	vector<TreeNode*> result;
	if(start>end) {
		result.push_back(NULL); //对应上一轮没有孩子节点的情况
		return result;}
	for(int i=start;i<=end;i++){ //依次选择不同数作为根节点
		//TreeNode *root=new TreeNode (i);  //此处易错,创建节点应在下面进行
		vector<TreeNode*> left;
		vector<TreeNode*> right;
		left=generate(start,i-1); //得到左子树(有多个子树)
		right=generate(i+1,end);  //得到右子树(有多个子树)
		for(int m=0;m<left.size();m++){  //遍历子树,将结果加入到最终结果中
			for(int n=0;n<right.size();n++){
                                TreeNode *root=new TreeNode (i);  //创建节点
				root->left=left[m];
				root->right=right[n];
				result.push_back(root);
			}
		}
	}
	return result;
}
    vector<TreeNode*> generateTrees(int n) {
        if(n<=0){   //对应n=0的情况,易错,若不加此段,n=0时结果将是[[]],而非[];
            vector<TreeNode*> result;
            return result;
        }
       return generate(1,n); 
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值