c++ - 使用STL算法计算绝对值之和

Question

我想使用std::numeric算法来计算数组的绝对值之和，以便使用gnu并行扩展(数组大小> 500000)。

这是我当前的代码:

double ret = 0;
for (auto i = 0U; i < length; ++i)
{
    ret += std::abs(tab[i]);
}
return ret;

所以我考虑这样做:

auto sumabs = [] (double a, double b) 
{
    return std::abs(a) + std::abs(b);
} 

std::accumulate(tab, tab + length, 0, sumabs);

但它效率低下，因为如果执行归约算法(我真诚地希望为了快速计算!)，std::abs 将应用于已经 >= 0 的值.

那么有什么办法可以做到这一点吗？也许“手动”执行减少的第一步，然后让 std::accumulate 在其余部分之间进行简单的加法？但会有拷贝和内存命中...

Answer 1

您可以将函数传递给 Accumlate 方法并在函数内执行“手动”计算。顺便说一句，在您的代码中，您将 abs 方法应用于第一个参数，这是不必要的。

int fAccumulate (int accumulated, int accumulateIncrement) 
{
    int retValue = 0;
    if (accumulateIncrement >= 0)
    {
        retValue = accumulated + accumulateIncrement;
    }
    else
    {
        retValue = accumulated + std::abs(accumulateIncrement);
    }
    return retValue;
}

此代码的用途可能是:

int init = 0;
int numbers[] = {10,20,-30};
int a = std::accumulate (numbers, numbers+3, init, fAccumulate);