Skip to content

Instantly share code, notes, and snippets.

@mkrueger
Created July 16, 2015 20:24
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save mkrueger/d73bde4a23cb385c9b5c to your computer and use it in GitHub Desktop.
Save mkrueger/d73bde4a23cb385c9b5c to your computer and use it in GitHub Desktop.
public class binarytrees {
public static void main(String[] args) throws Exception {
int n = args.length > 0 ? Integer.parseInt(args[0]) : 0;
int minDepth = 4;
int maxDepth = Math.max(minDepth + 2, n);
int stretchDepth = maxDepth + 1;
int check = (TreeNode.create(0, stretchDepth)).check();
System.out.println("stretch tree of depth " + (maxDepth + 1) + "\t check: " + check);
TreeNode longLivedTree = TreeNode.create(0, maxDepth);
for (int depth = minDepth; depth <= maxDepth; depth += 2)
{
int iterations = 1 << (maxDepth - depth + minDepth);
check = 0;
for (int i = 1; i <= iterations; i++)
{
check += (TreeNode.create(i, depth)).check();
check += (TreeNode.create(-i, depth)).check();
}
System.out.println((iterations << 1) + "\t trees of depth " + depth + "\t check: " + check);
}
System.out.println("long lived tree of depth " + maxDepth + "\t check: " + longLivedTree.check());
}
static class TreeNode {
int item;
TreeNode left, right;
static TreeNode create(int item, int depth)
{
return ChildTreeNodes(item, depth - 1);
}
static TreeNode ChildTreeNodes(int item, int depth)
{
TreeNode node = new TreeNode(item);
if (depth > 0)
{
node.left = ChildTreeNodes(2 * item - 1, depth - 1);
node.right = ChildTreeNodes(2 * item, depth - 1);
}
return node;
}
TreeNode(int item) {
this.item = item;
}
int check() {
return left == null ? item : left.check() - right.check() + item;
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment