1011. 复数类

http://acm.sjtu.edu.cn/OnlineJudge/problem/1011

学习了C++重载的写法。

# include <iomanip>
# include <iostream>

using namespace std;

class MyComplex{
    double real;
    double imag;

    friend istream &operator>>(istream & input, MyComplex & z) {
        input >> z.real >> z.imag ;
        return input;
    }
    friend ostream &operator<<(ostream & output, const MyComplex & z) {
        output.precision(2);
        output << fixed << z.real << ' ' << fixed << z.imag ;
        return output;
    }

public:
//    MyComplex(){
//        real = 0;
//        imag = 0;
//    }
    MyComplex(double real = 0, double imag = 0) {
        this->real = real;
        this->imag = imag;
    }
    const MyComplex operator+(MyComplex &);
    const MyComplex operator-(MyComplex &);
    const MyComplex operator*(MyComplex &);
    const MyComplex operator/(MyComplex &);
    const MyComplex operator+=(MyComplex &);
    const MyComplex operator-=(MyComplex &);
    const MyComplex operator*=(MyComplex &);
    const MyComplex operator/=(MyComplex &);
};

const MyComplex MyComplex::operator+(MyComplex &rz)
{
    MyComplex ret;
    ret.real = real + rz.real;
    ret.imag = imag + rz.imag;
    return ret;
}

const MyComplex MyComplex::operator-(MyComplex & rz)
{
    MyComplex ret;
    ret.real = real - rz.real;
    ret.imag = imag - rz.imag;
    return ret;
}

const MyComplex MyComplex::operator*(MyComplex & rz)
{
    MyComplex ret;
    ret.real = real*rz.real - imag*rz.imag;
    ret.imag = real*rz.imag + rz.real*imag;
    return ret;
}

const MyComplex MyComplex::operator/(MyComplex & rz)
{
    MyComplex ret;
    double m2 = rz.real*rz.real + rz.imag*rz.imag;
    ret.real = real*rz.real + imag*rz.imag;
    ret.imag = rz.real*imag - real*rz.imag;
    ret.real /= m2;
    ret.imag /= m2;
    return ret;
}

const MyComplex MyComplex::operator+=(MyComplex & rz)
{

    return (*this) = (*this) + rz;
}

const MyComplex MyComplex::operator-=(MyComplex & rz)
{
    return (*this) = (*this) - rz;
}

const MyComplex MyComplex::operator*=(MyComplex & rz)
{
    return (*this) = (*this) * rz;
}

const MyComplex MyComplex::operator/=(MyComplex & rz)
{
    return (*this) = (*this) / rz;
}

int main()
{
    MyComplex z1;
    MyComplex z2;

    cin >> z1 >> z2;

    cout << z1 + z2 << endl;
    cout << z1 - z2 << endl;
    cout << z1 * z2 << endl;
    cout << z1 / z2 << endl;
    cout << (z1 += z2) << endl;
    cout << (z1 -= z2) << endl;
    cout << (z1 *= z2) << endl;
    cout << (z1 /= z2) << endl;

    return 0;
}