蓝桥杯 算法提高 6-17 复数四则运算

算法提高 6-17复数四则运算  

时间限制：1.0s   内存限制：512.0MB

　　设计复数库，实现基本的复数加减乘除运算。
　　输入时只需分别键入实部和虚部，以空格分割，两个复数之间用运算符分隔；输出时按a+bi的格式在屏幕上打印结果。参加样例输入和样例输出。
　　注意考虑特殊情况，无法计算时输出字符串"error"。

样例输入

2 4 * -3 2

样例输出

-14-8i

样例输入

3 -2 + -1 3

样例输出

2+1i

 

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
struct num{
    double r,e;
};
num add(num a,num b){
    num c;
    c.r=a.r+b.r;
    c.e=a.e+b.e;
    return c;
}
num sub(num a,num b){
    b.r=-b.r;
    b.e=-b.e;
    return add(a,b);
}
num mul(num a,num b){
    num c;
    c.r=a.r*b.r-a.e*b.e;
    c.e=a.r*b.e+a.e*b.r;
    return c;
}
num div(num a,num b){
    num c;
    b.e=-b.e;
    c=mul(a,b);
    double m=b.e*b.e+b.r*b.r;
    c.r=c.r/m;
    c.e=c.e/m;
    return c;
}
int main(){
    //freopen("D:\\input.txt","r",stdin);
    num a,b,c;
    char cal;
    while(cin>>a.r>>a.e>>cal>>b.r>>b.e){//scanf("%f %f",&a.r,&a.e),cin>>cal,scanf("%f %f",&cal,&b.r,&b.e)){
        //cout<<cal<<endl;
        switch(cal){
        case '+':{
            c=add(a,b);
            cout<<c.r;
            if(c.e>0){
                cout<<"+";
            }
            cout<<c.e<<"i"<<endl;
            break;
        }
        case '-':{
            c=sub(a,b);
            cout<<c.r;
            if(c.e>0){
                cout<<"+";
            }
            cout<<c.e<<"i"<<endl;
            break;
        }
        case '*':{
            c=mul(a,b);
            cout<<c.r;
            if(c.e>0){
                cout<<"+";
            }
            cout<<c.e<<"i"<<endl;
            break;
        }
        case '/':{
            if(b.r==0&&b.e==0){
                cout<<"error"<<endl;

            }
            else{
                c=div(a,b);
                cout<<c.r;
                if(c.e>0){
                    cout<<"+";
                }
                cout<<c.e<<"i"<<endl;
            }
            break;
        }
        }
    }
    return 0;
}