demo_operator_07_a.cpp

来自「对于一个初涉VC＋＋的人来书」· C++ 代码 · 共 45 行

//************************************************
// 重载流插入运算符"<<"和流提取运算符">>"
//************************************************

# include <iostream.h>

class Complex
{
public:
    friend ostream& operator << (ostream&,Complex&); //输出流类
    friend istream& operator >> (istream&,Complex&); //输入流类
private:
    double real;
    double imag;
};
   
ostream& operator << (ostream& output,Complex& c)
{
	output<<"("<<c.real;
	if(c.imag>=0) output<<"+";
	output<<c.imag<<"i)";

	return output;
}

istream& operator >> (istream& input,Complex& c)
{
	cout<<"input real part and imaginary part of complex number:";
	input>>c.real>>c.imag;

	return input;
}

int main()
{
	Complex c1,c2;
	cin>>c1>>c2;
	cout<<"c1="<<c1<<endl;
	cout<<"c2="<<c2<<endl;

	return 0;
}