95. Unique Binary Search Trees II
Medium
1552129FavoriteShare
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
Accepted
157,746
Submissions
421,690
这道题是真的很不错,用了递归,到达递归边界时也要push进去一个NULL,NULL也算作一个元素,这个以前忽略掉了
class Solution {
public:
vector<TreeNode*> generateTrees(int n) {
vector<TreeNode*> rt;//存的是每棵树的根节点指针
if(n==0) return rt;
dfs(1,n,rt);//dfs递归建树
return rt;
}
void dfs(int start,int end,vector<TreeNode*>& treesroot){
if(start>end){
treesroot.push_back(NULL);
return;
}
for(int i=start;i<=end;i++){//依次设为根节点建树
vector<TreeNode*> treesleft;
vector<TreeNode*> treesright;
dfs(start,i-1,treesleft);//构建左子树。左子树为空即1为根时treesleft.size=1
dfs(i+1,end,treesright);//构建右子树
for(int j=0;j<treesleft.size();j++){//排列组合左右子树
for(int k=0;k<treesright.size();k++){
TreeNode* root=new TreeNode(i);
root->left=treesleft[j];
root->right=treesright[k];
treesroot.push_back(root);
}
}
}
}
};