C ++将常量添加到矢量切片

Question

我知道使用transform我可以为某些向量添加一个常量，如下所示：

std::vector<int> a(3, 2);
std::transform( a.begin(), a.end(), a.begin(), std::bind2nd( std::plus<double>(), 1 ) );

我想知道如何修改transform以向矢量的某些切片[index:end]添加常数，例如最后两个要素。

我可以用循环例如：

for (int i=1; i < a.size(); i++) {
    a.at(i) += 1;
}

但也许有更好的选择

Answer 1

只是最后一个N元素？使用反向迭代器：

#include <algorithm>
#include <iostream>
#include <iterator>
#include <vector>

int main() {
  std::vector<int> a(3, 2);
  std::transform(a.rbegin(), std::next(a.rbegin(), 2), a.rbegin(),
                 [](auto n) { return n + 1; });

  for (auto n : a) {
    std::cout << n << '\n';
  }
  return 0;
}

---

（如果由于某种原因需要将std::next()更改为没有随机迭代器的容器类型，则使用a.rbegin() + 2而不仅仅是a会更容易。）

Answer 2

算法采用迭代器而不是容器的原因正是这样的情况。

调整Python样式切片的一般方法是将+ve值添加到begin，将-ve添加到end。假设至少有两个元素，则递增最后两个元素

auto start = a.end() - 2;
auto finish = a.end();
std::transform( start, finish , start, std::bind2nd( std::plus<double>(), 1 ) );

Answer 3

小型重构后的相同代码：

std::vector< int > a(3, 2);

auto start = a.begin();
auto end   = a.end();
auto func  = [](auto val) {
    return val + 1;
};

std::transform(start, end, start, func);

现在通过修改start和end的值，只能修改切片。

为此，可以使用std::advance：

std::advance(start, 1);
std::advance(end, -1);

在上面的代码中：

• start在开头之后是1个元素（第2个元素 - 索引为1的元素）。
• end是最后的1个元素。

Answer 4

您可以使用跨度（早期的C ++ 20中的std::span，gsl::span）来表示切片。如果您不了解跨度，请查看：

What is a "span" and when should I use one?

所以，如果你写，

// ... vector a gets defined
auto slice = std::span(a).subspan(some_index);

切片现在的行为与任何标准库容器一样。具体来说，你可以写：

std::transform(slice.begin(), slice.end(), slice.begin(), [](auto v) { return v + 1; });