add values in arrays with positive and zero values

Hi,

I have a two vectors a=[1 2 3 3 3], b=[0 0 1 0 2], I want to add them in such a way in order to get

c=[1 2 4 4 6] which means that i just add them if b=0 and if it's not I calculate the sum and put the result also in the next value for a. Then i add the next b

I don't know if it's clear

thanks!!

Answer 1

[Andrei Bobrov]

a=[1 2 3 3 3];
b=[0 0 1 0 2];

c = a + cumsum(b);

[Nikolas Spiliopoulos]

thank you sir!