在 的限制下 matlab,matlab – 有上限和下限的cumsum？

我想找到一种矢量化方法来计算向量的累积和,但是有上限和下限.

在我的例子中,输入只包含1和-1.您可以在答案中使用此假设.当然,也欢迎更通用的解决方案.

例如：

x = [1 1 1 1 -1 -1 -1 -1 -1 -1];

upper = 3;

lower = 0;

s = cumsum(x) %// Ordinary cumsum.

s =

1 2 3 4 3 2 1 0 -1 -2

y = cumsumlim(x, upper, lower) %// Cumsum with limits.

y =

1 2 3 3 2 1 0 0 0 0

^ ^

| |

upper limit lower limit

当累积总和达到上限(在第3个元素)时,它将不再增加.同样,当累积总和达到下限(在第7个元素)时,它不会再减少. for循环版本将是这样的：

function y = cumsumlim(x, upper, lower)

y = zeros(size(x));

y(1) = x(1);

for i = 2 : numel(x)

y(i) = y(i-1) + x(i);

y(i) = min(y(i), upper);

y(i) = max(y(i), lower);

end

end

你有什么想法？