在数组中添加值并与Matlab中循环内的阈值进行比较

Question

我一直试图弄清楚这一点。我有一个数组：

a = [ 1 1 1 2 1 1 1 3 2 1 1 2 1 1 1]

我想add the values in the array so that it equal to 10。一旦添加的值达到 10，我希望数组再次开始添加值，直到达到 10。我在这里面临两个问题，

1) 我怎样才能添加数组，以便 sum = 10 每次。请注意，在数组中，有3。如果我在3 之前添加所有值，我会得到8，我只需要来自3 的2。我需要确保将余数（即 1）添加到下一个数组中以获得总和 10。

2) 一旦循环到达10，我如何打破循环并要求它继续求和到下一个值以获得另一个10？

我创建了一个循环，但它只适用于数组的第一部分。我不知道如何让它继续。代码如下：

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

for i = 1:length(a) 
   while c < 10
      c = c + a(i);
   break
   end
end

请帮忙。谢谢

Answer 1

这可以使用cumsum、mod、diff 和find 来完成，如下所示：

temp = cumsum(a);
required = find([0 diff(mod(temp,10))] <0)

cumsum 返回累积和，然后使用mod 重新调整。 diff 确定总和大于或等于 10 的位置，最后find 确定这些索引。

编辑： 如果a 没有负面元素，则上述解决方案有效。如果a 可以有否定元素，那么：

temp1=cumsum(a);              %Commulative Sum
temp2=[0 diff(mod(temp1,10))];%Indexes where sum >=10 (indicated by negative values)
temp2(temp1<0)=0;             %Removing false indexes which may come if `a` has -ve values
required = find(temp2 <0)     %Required indexes

Answer 2

这应该做你正在尝试的。它显示每次总和等于 10 的索引。用你的测试用例检查这个。 rem 存储每次迭代中的残差和，并在下一次迭代中结转。其余代码与您所做的类似。

a = [ 1 1 1 2 1 1 1 3 2 1 1 2 1 1 1]; 
c = 0; 
rem = 0;
i = 1;
length(a);
while(i <= length(a))
   c = rem; 
   while (c < 10 && i <= length(a))
      c = c + a(i);
      i = i + 1;
      if(c >= 10)
        rem = c - 10;
        break
      end
   end
   if(c >= 10)
      disp(i-1)
end

Answer 3

使用cumsum 而不是while 循环：

a = [ 1 1 1 2 1 1 1 3 2 1 1 2 1 1 1];
a_ = a;
endidxlist = false(size(a));
startidxlist = false(size(a));
startidxlist(1) = true;
while any(a_) && (sum(a_) >= 10)
    b = cumsum(a_);
    idx = find(b >= 10,1);
    endidxlist(idx) = true;
    % move residual to the next sequence
    a_(idx) = b(idx) - 10;
    if a_(idx) > 0
        startidxlist(idx) = idx;
    elseif (idx+1) <= numel(a)
        startidxlist(idx+1) = true;
    end
    a_(1:idx-1) = 0;
end
if (idx+1) <= numel(a)
    startidxlist(idx+1) = false;
end

endidxlist 为您提供每个序列的结束索引，startidxlist 为您提供开始索引