为STL Vector重载运算符

时间:2013-10-22 08:52:10

标签: c++ stl

我知道这通常不是最好的主意,但我决定尝试一下。我正在尝试重载+运算符以使元素明智地添加两个向量。由于某种原因,它只能正确返回向量中的第一个值。 我无法弄清楚出了什么问题?如果我在计算sum数组的元素时使用push_back而不是调整大小,它也无法工作。这是代码:

    #include "stdafx.h"
    #include <iostream>
    #include "conio.h"
    #include <vector>

    using namespace std;

    class dvector{

    public:
        vector<double> q;
        dvector(vector<double>);
        dvector();
    };
    dvector::dvector(vector<double> a) {
        q=a;
    };
    dvector::dvector() {
        vector<double> q;
    };

    dvector operator+(dvector c1, dvector c2)
    {
        int sz=c1.q.size();
        dvector sum;
        sum.q.resize(sz);
        double temp;
        for (int i=0; i<sz; i++){
            sum.q[i]=c1.q[i]+c2.q[i];
            return sum;
        }
    }


    int _tmain(int argc, _TCHAR* argv[])

    {
        vector<double> test1;
        test1.resize(10);
        for (int i=0; i<10;i++){
            test1[i]=i;
        }
        vector<double> test2;
        test2.resize(10);
        for (int i=0; i<10;i++){
            test2[i]=2;
        }
        dvector dtest1(test1); 
        dvector dtest2(test2); 
        dvector sum=dtest1+dtest2;

        for (int i =0;i<10;i++)

            {
                cout<<sum.q[i]<<' ';
            }
        getch();
        return 0;
    }

2 个答案:

答案 0 :(得分:5)

return中的for - 在第一次迭代后立即循环。您应该将return sum;移到operator+的最后一行。

答案 1 :(得分:4)

因为你太快回来了。 return语句不仅会突破for loop,还会退出功能。

dvector operator+(dvector c1, dvector c2)
{
    int sz=c1.q.size();
    dvector sum;
    sum.q.resize(sz);
    double temp;
    for (int i=0; i<sz; i++){
        sum.q[i]=c1.q[i]+c2.q[i];
        //return sum;  // don't return until you finish calculating
    }
    return sum;    // move to here
}