抓住一个“子” - 向量和“连接”向量

时间:2012-03-30 18:22:11

标签: c++ boost mpi boost-mpi

所以我正在编写一个并行程序(boost.mpi),我希望传递一个向量的片段(如在std :: vector中),它可以被组合成一个完整的向量。

为了做到这一点,我希望能够做两件事:

  1. 抓取一个向量 - 即说一个向量有800个元素,那么最好的方法是创建一个包含元素200-299的子向量(或由2个int变量定义的任意索引)?

  2. 我想创建一个运算符,允许我将向量相加以创建一个新的更长的向量。基本上,我想要与std :: plus()(可以连接字符串)相同的功能,但是对于矢量。我需要能够将此运算符作为二元运算符传递(它需要与std :: plus()具有相同的类型)。

  3. 对此有任何帮助将非常感激。

1 个答案:

答案 0 :(得分:0)

第一部分可以通过使用以下向量构造函数来实现:

template <class InputIterator>
vector( InputIterator first, InputIterator last, 
        const Allocator& alloc = Allocator() );

第二部分可以使用vector::insert来实现,但可能有更好的方法。我已经给出了下面各自的样本。

#include <vector>
using std::vector;

template <typename T>
vector<T> operator+ (const vector<T> &lhs, const vector<T> &rhs)
{
    vector<T> ret (lhs);
    ret.insert (ret.end(), rhs.begin(), rhs.end());
    return ret;
}

/// new //create a structure like std::plus that works for our needs, could probably turn vector into another template argument to allow for other templated classes
template <typename T>
struct Plus
{
    vector<T> operator() (const vector<T> &lhs, const vector<T> &rhs)
    {
        return lhs + rhs;
    }
};
/// end new

#include <iostream>
using std::cout;

/// new
#include <numeric>
using std::accumulate;
/// end new

template <typename T>
void print (const vector<T> &v)
{
    for (const T &i : v)
        cout << i << ' ';

    cout << '\n';
}

int main()
{
    vector<int> vMain {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; //syntax only available in C++11
    vector<int> vSub (vMain.begin() + 3, vMain.begin() + 6); //vMain[0+3]=4, vMain[0+6]=7
    vector<int> vCat = vMain + vSub;

    /// new
    vector<vector<int>> vAdd {vMain, vMain}; //create vector of vector of int holding two vMains
    /// end new

    print (vMain);
    print (vSub);
    print (vCat);

    /// new //accumulate the vectors in vAdd, calling vMain + vMain, starting with an empty vector of ints to hold the result
    vector<int> res = accumulate (vAdd.begin(), vAdd.end(), (vector<int>)(0));//, Plus<int>());
    print (res); //print accumulated result
    /// end new
}

输出:

1 2 3 4 5 6 7 8 9 10
4 5 6
1 2 3 4 5 6 7 8 9 10 4 5 6
1 2 3 4 5 6 7 8 9 10 1 2 3 4 5 6 7 8 9 10

编辑: 我对如何做到这一点真的很不满意,但我已经更新了代码以处理std::accumulate之类的内容。