复数运算(c++

#include <bits/stdc++.h>

// 定义一个结构体类型complex
struct complex {
    double real;
    double imag;
};

// 函数声明
complex add(complex x1, complex x2);  // 两个复数相加，返回一个复数
complex sub(complex x1, complex x2);  // 两个复数相减，返回一个复数
void output(complex x);               // 输出复数

int main() 
{
    complex c1{3.0, 4.0}, c2{2.1, 5.5};  // 定义两个结构体变量，并初始化
    complex c3, c4;  // 定义两个结构体变量

    c3 = add(c1, c2); 
    c4 = sub(c1, c2);

    output(c3);
    output(c4);


}

// 函数定义
// 两个复数相加，返回一个复数
complex add(complex x1, complex x2) 
{
    complex result;
    
    result.real = x1.real + x2.real;
    result.imag = x1.imag + x2.imag;

    return result;
}


// 两个复数相减，返回一个复数
complex sub(complex x1, complex x2) 
{
    complex result;
    
    result.real = x1.real - x2.real;
    result.imag = x1.imag - x2.imag;

    return result;
}

// 输出复数
void output(complex x) 
{
    if(x.imag >= 0)
        std::cout << x.real << "+" << x.imag << "i" << std::endl;
    else
        std::cout << x.real << x.imag << "i" << std::endl;

}