let mapi f a = Array.mapi (fun i v -> f i v) a