improve: Reduce gas and minimize overflow risk in for loops

[nicholaspai]

Leaving the counter i variable as uint256 removes gas cost to cast to uint32 and also prevents chance that leaf count will cause overflow, which is a risk with the increment logic ++i being inside an unchecked block.