matlab 累积和,区间累积求和 – MATLAB

diff和cumum的一种方法 –

%// Setup few arrays:

cx = cumsum(x) %// Continuous Cumsumed version

reset_mask = reset==1 %// We want to create a logical array version of

%// reset for use as logical indexing next up

%// Setup ID array of same size as input array and with differences between

%// cumsumed values of each group placed at places where reset==1, 0s elsewhere

%// The groups are the islands of 0s and bordered at 1s in reset array.

id = zeros(size(reset))

diff_values = x(reset_mask) - cx(reset_mask)

id(reset_mask) = diff([0 diff_values])

%// "Under-compensate" the continuous cumsumed version cx with the

%// "grouped diffed cumsum version" to get the desired output

y = cx + cumsum(id)