binary-trees Ruby yjit #3 program

# The Computer Language Benchmarks Game
# https://salsa.debian.org/benchmarksgame-team/benchmarksgame/
#
# contributed by Jesse Millikan
# Modified by Wesley Moxam and Michael Klaus
# Modified by  Chris Houhoulis:
#   trigger GC; move stretch_tree into loop; use while instead of a block
#  *reset* 

def item_check(left, right)
  return 1 if left.nil?
  1 + item_check(*left) + item_check(*right)
end

def bottom_up_tree(depth)
  return [nil, nil] if depth == 0
  depth -= 1
  [bottom_up_tree(depth), bottom_up_tree(depth)]
end

max_depth = ARGV[0].to_i
min_depth = 4

max_depth = [min_depth + 2, max_depth].max

1.times do
  stretch_depth = max_depth + 1
  stretch_tree = bottom_up_tree(stretch_depth)
  puts "stretch tree of depth #{stretch_depth}\t check: " +
    "#{item_check(*stretch_tree)}"
end
GC.start

long_lived_tree = bottom_up_tree(max_depth)

base_depth = max_depth + min_depth
min_depth.step(max_depth, 2) do |depth|
  iterations = 2 ** (base_depth - depth)

  check, i = 0, 1

  while i <= iterations
    temp_tree = bottom_up_tree(depth)
    check += item_check(*temp_tree)
    i += 1
  end

  puts "#{iterations}\t trees of depth #{depth}\t check: #{check}"
end
GC.start

puts "long lived tree of depth #{max_depth}\t check: " +
  "#{item_check(*long_lived_tree)}"
    

NOTES:
64-bit Ubuntu quad core
ruby 3.3.5
(2024-09-03
revision ef084cc8f4)
+YJIT [x86_64-linux]


 Wed, 04 Sep 2024 18:11:05 GMT

COMMAND LINE:
 /opt/src/ruby-3.3.5/bin/ruby --yjit -W0 binarytrees.ruby-3.ruby 21

PROGRAM OUTPUT:
stretch tree of depth 22	 check: 8388607
2097152	 trees of depth 4	 check: 65011712
524288	 trees of depth 6	 check: 66584576
131072	 trees of depth 8	 check: 66977792
32768	 trees of depth 10	 check: 67076096
8192	 trees of depth 12	 check: 67100672
2048	 trees of depth 14	 check: 67106816
512	 trees of depth 16	 check: 67108352
128	 trees of depth 18	 check: 67108736
32	 trees of depth 20	 check: 67108832
long lived tree of depth 21	 check: 4194303