Files
rose-ash/lib/ocaml/baseline/fenwick_tree.ml
giles cb626fc402
Some checks failed
Test, Build, and Deploy / test-build-deploy (push) Failing after 25s
ocaml: phase 5.1 fenwick_tree.ml baseline (BIT over 8 elements, fingerprint 228)
Fenwick / Binary Indexed Tree for prefix sums. The classic
`i & -i` low-bit trick needs negative-aware AND, but our `land`
evaluator (iter 127, bitwise via floor/mod arithmetic) only handles
non-negative operands. Workaround: a portable lowbit helper that
finds the largest power of 2 dividing i:

  let lowbit i =
    let r = ref 1 in
    while !r * 2 <= i && i mod (!r * 2) = 0 do
      r := !r * 2
    done;
    !r

After building from [1;3;5;7;9;11;13;15]:
  total       = prefix_sum 8 = 64
  update 1 by +100
  after       = prefix_sum 8 = 164
  total + after = 228

Tests recursive update / prefix_sum chains via helper-extracted
lowbit; documents a non-obvious limit of the bitwise-emulation
layer.

168 baseline programs total.
2026-05-10 23:48:46 +00:00

34 lines
594 B
OCaml

let n = 8
let bit = Array.make (n + 1) 0
let lowbit i =
let r = ref 1 in
while !r * 2 <= i && i mod (!r * 2) = 0 do
r := !r * 2
done;
!r
let rec update i delta =
if i <= n then begin
bit.(i) <- bit.(i) + delta;
update (i + lowbit i) delta
end
let rec prefix_sum i =
if i <= 0 then 0
else bit.(i) + prefix_sum (i - lowbit i)
let range_sum l r = prefix_sum r - prefix_sum (l - 1)
;;
let a = [| 1; 3; 5; 7; 9; 11; 13; 15 |] in
for i = 0 to n - 1 do
update (i + 1) a.(i)
done;
let total = prefix_sum n in
update 1 100;
let after = prefix_sum n in
total + after