流插入运算符和流提取运算符重载为友元(P165)

/*

在C++中,左移运算符“<<”可以和cout一起用于输出,称为“流插入运算符”。
右移运算符“>>”和cin一起用于输入,称为流提取运算符。

重载流插入运算符的格式:
ostream &operator<<(ostream & output,类名&对象名)
{
~
return output;
}

重载流提取运算符的格式:
istream &operator>>(istream &input,类名&对象名)
{
~
return input;
}

*/

#include<iostream>
#include<string>
#include<cstdlib>
using namespace std;

class myComplex
{
private:
double real,imag;
public:
myComplex():real(0),imag(0){}
myComplex(double r,double i):real(r),imag(i){}
friend ostream & operator<<(ostream & os,const myComplex & c); //友元,插入
friend istream & operator>>(istream & is,myComplex & c); //友元,提取
};

ostream & operator<<(ostream & os,const myComplex & c)
{
if(c.imag>=0)
os<<c.real<<"+"<<c.imag<<"i"; //以a+bi的形式输出
else
os<<c.real<<"+"<<(-c.imag)<<"i";
return os;
}

istream & operator>>(istream & is,myComplex & c)
{
string s;
is>>s; //将a+bi作为字符串读入,a+bi中间不能有空格
int pos = s.find("+",0); //查找虚部
if(pos==-1)
pos=s.find("-",1); //虚部为负数时
string sReal=s.substr(0,pos); //分离出代表实部的字符串
c.real=atof(sReal.c_str()); //atof()能将参数内容转换成浮点数
sReal=s.substr(pos,s.length()-pos-1); //分离出代表虚部的字符串
c.imag=atof(sReal.c_str());
return is;
}

int main()
{
myComplex c,c1;
int n;
cout<<"请输入两个复数([-]a+-bi)和一个整数,以空格分隔"<<endl;
cin>>c>>c1>>n;
cout<<c<<","<<n<<","<<c1<<endl;
return 0;
}

 

posted @ 2020-03-15 10:12  CollisionDimension  阅读(515)  评论(0)    收藏  举报