aboutsummaryrefslogtreecommitdiff
path: root/docs/history/17-o3-boundaries/code/examples/tree.zc
blob: 90a01d4b1f4590094702e362b366936e441dc001 (plain) (blame)
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
// a binary search tree where every node owns its children. there's no free
// anywhere: when the root's block ends, the whole tree goes with it.
struct node { int v; *node left, right; };

*node.u leaf(int v) { *node n = new(node); n.v = v; return u*n; }
// n is borrowed; storing a new leaf into it makes n the leaf's owner
fn insert(*node.m n, int v)
{
	if (v < n.v) { if (n.left) insert(n.left, v); else n.left = leaf(v); }
	else { if (n.right) insert(n.right, v); else n.right = leaf(v); }
	return 0;
}
fn walk(*node n) { if (!n) return 0; walk(n.left); putn(n.v); putc(' '); walk(n.right); return 0; }
int height(*node n) { if (!n) return 0; int l = height(n.left), r = height(n.right); return 1 + (l > r ? l : r); }

fn main()
{
	let before = live_objects();
	{
		*node root = leaf(50);
		int x = 7;
		for (int i = 0; i < 30; i++) { x = (x * 37 + 11) % 100; insert(root, x); }
		walk(root); putc(10);
		puts("height "); putn(height(root)); puts(", live objects "); putn(live_objects() - before); putc(10);
	}
	puts("after the block: "); putn(live_objects() - before); putc(10);
	return 0;
}