如何遍历通过引用传递的多个对象?

假设我们有这个函数,其中 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;
}

没有深拷贝。


以上是如何遍历通过引用传递的多个对象?的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>