МАТЛАБ: Как найти число, соответствующее определенному условию, и пометить его как «0»?

Question

Например,

A = [1 0 1 1 0 -1 0 1 -1]

for i = 1: length(A)

как пометить число как "0", когда sum(A(1:i)) ~= 1 && sum(A(1:i)) ~= 0?

а как написать код?

Answer 1

A = [1 0 1 1 0 -1 0 1 -1]

%# create all sums from 1 through i
sa = cumsum(A);

%# in output: ith element is true if sum from 
%# 1 through the ith element in A is 0 or 1
output = ismember(sa,[0 1])

output = 
   1     1     0     0     0     0     0     0     0