資工趴趴熊的小天地

問題描述 ：

在做傅立葉轉換時，常會用到複數，但每次都要分開來計算實部與虛部，非常的麻煩，現在透過operator overloading的方式來簡化程式設計師的負擔。須做加減乘。

輸入說明 ：

第一列輸入一個正整數n。其後有n列，每一列代表一個想要做運算的虛數，每一列之資料依序為運算元、虛數1、虛數2。虛數的格式為a b。

輸出說明 ：

每一列表一個運算結果。虛數的格式為a b。

範例 ：

C++

#include<iostream>
using namespace std;

int main(){
    int n;
    cin>>n;
    while (n--)
    {
        int a1,a2,b1,b2;
        char w;
        cin>>w>>a1>>a2>>b1>>b2;
        switch(w){
            case '+':cout<<a1+b1<<" "<<a2+b2<<"\n";break;
            case '-':cout<<a1-b1<<" "<<a2-b2<<"\n";break;
            case '*':cout<<a1*b1-a2*b2<<" "<<a2*b1+a1*b2<<"\n";break;
            case '/':cout<<(a1*b1+a2*b2)/(b1*b1+b2*b2)<<" "<<(a2*b1-a1*b2)/(b1*b1+b2*b2)<<"\n";break;
        }
    }
    
}

python:

while True:  
    try:  
        n=int(input())  
        for _ in range(n):  
            w,a1,a2,b1,b2=map(str,input().split())  
            a1=int(a1)  
            a2=int(a2)  
            b1=int(b1)  
            b2=int(b2)  
            if w=='+':print(a1+b1,a2+b2)  
            elif w=='-':print(a1-b1,a2-b2)  
            elif w=='*':print(a1*b1-a2*b2,a2*b1+a1*b2)  
            else:print((a1*b1+a2*b2)/(b1*b1+b2*b2),(a2*b1-a1*b2)/(b1*b1+b2*b2))  
    except:break