95. Unique Binary Search Trees II
recursion or dp
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<TreeNode> generateTrees(int n) {
// start->end
return recursion(1,n);
}
private List<TreeNode> recursion(int start, int end){
List<TreeNode> list = new LinkedList<>();
if (start>end){
list.add(null);
return list;
}
for(int i=start; i<=end; i++){
List<TreeNode> leftList= recursion(start, i-1);
List<TreeNode> rightList = recursion(i+1, end);
for(TreeNode leftNode: leftList){
for(TreeNode rightNode: rightList){
TreeNode node=new TreeNode(i);
node.left = leftNode;
node.right = rightNode;
list.add(node);
}
}
}
return list;
}
}
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<TreeNode> generateTrees(int n) {
List<TreeNode>[] ans = new List[n+1];
ans[0]=new LinkedList<>();
ans[0].add(null);
ans[1]=new LinkedList<>();
ans[1].add(new TreeNode(1));
for(int i=2; i<=n; i++){
ans[i]= new LinkedList<>();
// node(j) is the root
for(int j=1; j<=i; j++){
List<TreeNode> left = ans[j-1];
// right mirrors the left part
// 1. number of right nodes = i-j;
// 2. the value of right node = original left node val + rootNode val
List<TreeNode> right = ans[i-j];
for(TreeNode nodeL: left){
for(TreeNode nodeR: right){
TreeNode node = new TreeNode(j);
node.left = nodeL;
node.right = generateRightCopy(nodeR, j);
ans[i].add(node);
}
}
}
}
return ans[n];
}
private TreeNode generateRightCopy(TreeNode node, int offset){
if (node==null) return null;
TreeNode newNode = new TreeNode(node.val+offset);
newNode.left = generateRightCopy(node.left, offset);
newNode.right = generateRightCopy(node.right, offset);
return newNode;
}
}
Last updated
Was this helpful?