1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
|
public class UniqueBinarySearchTreesII { public List<TreeNode> generateTrees(int n) { return helper(1, n); } private List<TreeNode> helper(int start, int end) { List<TreeNode> result = new ArrayList<TreeNode>(); if (start > end) { result.add(null); return result; } for (int i = start; i <= end; i++) { List<TreeNode> left = helper(start, i - 1); List<TreeNode> right = helper(i + 1, end); for (TreeNode l : left) { for (TreeNode r : right) { TreeNode root = new TreeNode(i); root.left = l; root.right = r; result.add(root); } } } return result; } }
|