c++ - 是否可以使用 STL 算法重写我的代码？

Question

我有一个嵌套循环，我希望使用 STL 算法重写它。有人可以帮我吗？

std::bitset<array_size * 8> bitset {};
short bitsetIndex {0};
for (int i = 0; i < array_size; ++i) {
    std::bitset<8> cBitset { (unsigned char) charArray[i] };
    for (int i = 0; i < cBitset.size(); ++i) {
        if(cBitset.test(i)) {
            bitset.set(bitsetIndex);
        }
        bitsetIndex++;
    }
}
return bitset;

标签： c++stl-algorithm

Answer 1

看起来您想表示charArray为位的集合。一个简单的循环和几个位操作就可以完成这项工作：

using Bitset = std::bitset<array_size * 8>;
Bitset bitset;
for (auto i = array_size; i > 0; --i) {
    bitset <<= 8;
    bitset |= Bitset(static_cast<unsigned char>(charArray[i - 1]));
}
return bitset;

根据charArray具体情况，循环也可以用迭代器重写：

using Bitset = std::bitset<array_size * 8>;
Bitset bitset;
for (auto it = std::rbegin(charArray); it != std::rend(charArray); ++it) {
    bitset <<= 8;
    bitset |= Bitset(static_cast<unsigned char>(*it));
}
return bitset;

最后，如果你真的想使用一些标准库算法，你可以使用std::accumulate：

using Bitset = std::bitset<array_size * 8>;
return std::accumulate(std::rbegin(charArray), std::rend(charArray), Bitset{}, 
    [](Bitset bs, auto ch) { return (bs << 8) | 
        std::bitset<array_size * 8>(static_cast<unsigned char>(ch));
});

演示