python重载练习之复数加减法

解题思路:

注意事项:

参考代码:

class Complex:
   def __init__(self,real=0,imag=0):
       self.real=real
       self.imag=imag
   def __add__(self,other):
           return Complex(self.real+other.real,self.imag+other.imag)
   def __sub__(self,other):
           return Complex(self.real-other.real,self.imag-other.imag)
   def __str__(self):
           if self.real==0:
               if self.imag==0:
                   return "0"
               elif self.imag>0:
                   return f"{self.imag}i"
               else:
                   return f"-{abs(self.imag)}i"
           else:
               if self.imag==0:
                   return f"{self.real}"
               elif self.imag > 0:
                   return f"{self.real}+{self.imag}i"
               else:
                   return f"{self.real}{self.imag}i"

if __name__=="__main__":
   real1, imag1, operator, real2, imag2 = input().split()
   # 将输入的字符串转换为整数
   real1, imag1, real2, imag2 = map(int, [real1, imag1, real2, imag2])
   Complex1=Complex(real1,imag1)
   Complex2=Complex(real2,imag2)
   result=None
   if operator == "+":
       result= Complex1 + Complex2  # 执行加法
   elif operator == "-":
       result= Complex1 - Complex2
   print(result)