исходном массиве

Есть массив из нулей и единиц, мне нужно за один проход по массиву найти сумму единиц до каждого из нулей и результат записать в новую коллекцию, проблема в том, что в новую коллекцию не добавляется последний элемент, как мне это исправить? Спасибо