STL 向量的重载运算符

Overloading operators for STL Vector

本文关键字:运算符 重载 向量 STL      更新时间:2023-10-16

我知道这通常不是最好的主意,但我决定试一试。 我正在尝试重载 + 运算符以对两个向量进行元素明智添加。出于某种原因,它只正确返回向量中的第一个值。我不知道出了什么问题?如果我在计算 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;
    }

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

因为你回来得太快了。 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
}