使用std :: transform将向量(A)的向量添加到另一个向量(B)的向量

时间:2018-10-04 14:57:14

标签: c++ vector sum overloading addition

我对使用向量和对C ++进行一般的编码还很陌生,但仍未完全掌握该语言。我的询问如下:

    1.我的主要问题似乎是我的 transform 行,为什么会这样?
    2.如何打印A和B的矢量和?
3.如何重载[] []运算符以进行访问并使它正常工作? (即,如果编写了Mat [1] [3] = 4,则代码​​仍然可以正常工作)

#include <iostream> 
#include <algorithm>
#include <vector> 
#include <functional>

using namespace std;

class Matrix
{
public:
    double x;
    vector<vector<double> > I{ { 1, 0, 0, 0 },
    { 0, 1, 0, 0 },
    { 0, 0, 1, 0 },
    { 0, 0, 0, 1 } };

    vector<vector<double> > Initialization(vector<vector<double> > I, double x);
    vector<vector<double> > Copy(vector<vector<double> > I);
    void Print(vector<vector<double> > I);

};

vector<vector<double> > Matrix::Initialization(vector<vector<double> > I, double x)
{
    for (int i = 0; i < I.size(); i++) {
        for (int j = 0; j < I[i].size(); j++)
        {
            // new matrix 
            I[i][j] *= x;
        }
    }
    return I;
};

vector<vector<double> > Matrix::Copy(vector<vector<double> > I)
{
    vector<vector<double> > I_copy = I;
    return I_copy;
};

void Matrix::Print(vector<vector<double> > I)
{
    for (int i = 0; i < I.size(); i++) {
        for (int j = 0; j < I[i].size(); j++)
        {
            cout << I[i][j] << " ";
        }
        cout << endl;
    }
};


int main()
{
    Matrix m;
    vector<vector<double> > A;
    vector<vector<double> > B;

    cin >> m.x;

    A = m.Initialization(m.I, m.x);
    B = m.Copy(A);

    m.Print(A);
    m.Print(B);


    B.resize(A.size());


    transform(A.begin(), A.end(), B.begin(), A.begin(), plus<double>());

    return 0;
}


希望您能耐心帮助我修复代码并让我理解为什么我的语法不正确且无法编译。非常感谢<3

1 个答案:

答案 0 :(得分:0)

正如Jarod42在评论中指出的那样,您需要添加std::vector<double>的东西以及添加double的东西。

template <typename T>
std::vector<T> operator+(std::vector<T> lhs, const std::vector<T> & rhs)
{
    std::transform(lhs.begin(), lhs.end(), rhs.begin(), lhs.begin(), [](const T & a, const T & b){ return a + b; });
    return lhs;
}

请注意,我们复制了左侧,但仅引用了右侧。这也给我们提供了将结果写入的地方。

使用非常简单

int main()
{
    std::vector<std::vector<double> > A { { 0, 1 }, { 1, 0 } };
    std::vector<std::vector<double> > B { { 1, 0 }, { 0, 1 } };
    std::vector<std::vector<double> > C { { 1, 1 }, { 1, 1 } };

    std::cout << std::boolalpha << (A + B == C) << std::endl;
}

See it live!