将 valarray<bool> 转换为 valarray<int>

问题描述

我正在尝试将 valarray<bool> 转换为 0 和 1 的 valarray<int>。

我希望这样的事情会起作用（使用一元 + 调用积分提升，就像在单个 bool 上一样）：

#include <iostream>
#include <valarray>

int main() {
    std::valarray<int> xx = { 1,2,3,4,5 };
    std::valarray<int> yy = { 5,1 };
    std::valarray<int> zz = +(xx < yy);
    std::cout << zz[0] << std::endl;
    return 0;
}

似乎有些编译器是这样的，但有些则不是。那些不输出类似的东西：

<source>:7:24: error: no viable conversion from '_Expr<_UnClos<std::__unary_plus,std::_Expr,std::__detail::_BinClos<std::__less,_ValArray,int,int>>,std::_Expr<std::__detail::_BinClos<std::__less,int>,bool>::value_type>' (aka '_Expr<_UnClos<std::__unary_plus,bool>') to 'std::valarray<int>'
    std::valarray<int> zz = +(xx < yy);
                       ^    ~~~~~~~~~~

我可以使用“掩码索引”来写：

    std::valarray<int> zz = 0 * xx; zz[xx < yy] = 1;

但这有点麻烦，我想有一个单一的表达。

将单个 bool 转换为 int 很容易，但似乎 .apply() 不允许更改内容的类型，所以我不能简单地映射这样的数组上的转换。

有没有更简单的方法？

解决方法

首先，从 xx 中减去 yy 并让结果 valarray 使用 apply() 以便为每个元素返回 0 或 1：

std::valarray<int> zz = (xx - yy).apply([](int x) {return x < 0; });