[LeetCode] Unique Binary Search Trees II

生成不同结构的BST

最新推荐文章于 2020-12-09 16:51:01 发布

原创最新推荐文章于 2020-12-09 16:51:01 发布 · 683 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

74 篇文章

订阅专栏

本文介绍了一种算法，用于生成所有可能的不同结构的二叉搜索树（BST），这些树存储从1到n的值。举例来说，当n为3时，会生成5种不同的BST结构。文章提供了一个C++实现方案，通过递归方法解决该问题。

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

For example,
Given n = 3, your program should return all 5 unique BST's shown below.

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

/**
 * Definition for binary tree
 * 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) {
        // Note: The Solution object is instantiated only once and is reused by each test case.
        
		vector<TreeNode*> ret;
		if ( n==0 )
		{
		    ret.push_back(NULL);
		    return ret;
		}
		return solve(0,n);
    }
	vector<TreeNode*> solve(int start,int n)
	{
	    vector<TreeNode*> ret;
		if ( n==0 )
		{
		    ret.push_back(NULL);
		    return ret;
		}
		
		for(int left=0;left<n;left++)
		{
			int right=n-left-1;
			vector<TreeNode*> lTree=solve(start,left);
			vector<TreeNode*> rTree=solve(start+left+1,right);
			for(int i=0;i<lTree.size();i++)
				for(int j=0;j<rTree.size();j++)
				{
					TreeNode* root= new TreeNode(start+left+1);
					root->left=lTree[i];
					root->right=rTree[j];
					ret.push_back(root);
				}
		}
		return ret;
	}
};