如何遍历通过引用传递的多个对象?
假设我们有这个函数,其中 vec1 和 vec2 存储相同类型的数据:
int myFunc(vector<int>& vec1, vector<int>& vec2) {
for (auto const& elem : vec1) {
// do something
}
for (auto const& elem : vec2) {
// do the same thing
}
}
显然,重复代码并不是很好。但是,以下不是解决方案:
int myFunc(vector<int>& vec1, vector<int>& vec2) {
for (auto const& vec : {vec1, vec2}) {
for (auto const& elem : vec) {
// do something
}
}
}
这样做会导致 vec1 和 vec2 的深拷贝,这不是我们想要的。我在这里没有看到使用指针的方法,因为向量是通过引用传入的(假设我们不能更改函数签名)。
那么我们如何在不重复代码的情况下对 vec1 和 vec2 做同样的事情呢?
回答
一种解决方案是编写一个“做某事”的小 lambda,然后您不必多次重复该代码:
int myFunc(vector<int>& vec1, vector<int>& vec2)
{
auto do_something = [&](auto const & elem) {
// do something
}
for (auto const& elem : vec1) {
do_something(elem);
}
for (auto const& elem : vec2) {
do_something(elem);
}
}
如果你可以使用 range-v3 库,你可以这样写:
int myFunc(vector<int>& vec1, vector<int>& vec2)
{
for (auto const& elem : ranges::views::concat(vec1, vec2)) {
// dosomething
}
}
- I was going to pitch a function, but what, me worry?
回答
使用它std::reference_wrapper来避免制作深层副本相当简单,并且几乎可以做与您最初想做的相同的事情:
#include <iostream>
#include <functional>
#include <vector>
int myFunc(std::vector<int>& vec1, std::vector<int>& vec2) {
for (auto const& vec : {std::ref(vec1), std::ref(vec2)}) {
auto const &real_vec=vec.get();
for (auto const& elem : real_vec) {
}
}
return 0;
}
int main()
{
std::vector<int> vec1, vec2;
myFunc(vec1, vec2);
return 0;
}
没有深拷贝。