Suma incrementable hasta encontrar valor

Hola a todos.

Tengo una pregunta que os la explico con un ejemplo.

Tengo una columna con 1 y 0. Quiero que en otra columna cuando encuentre un 1 ponga 1 y vaya sumando 1 en las siguientes hasta que encuentre otro 1, entonces empezaría el proceso. Es decir, que quedaría así:

A B

1 1

0 2

0 3

0 4

1 1

0 2

0 3

Gracias!