Design prefix-sum function and max stack