python重载练习之复数加减法

解题思路:

注意事项:

参考代码:

class Complex:
    def __init__(self, real1, imag1, real2, imag2, op):
        self.real1 = real1
        self.imag1 = imag1
        self.real2 = real2
        self.imag2 = imag2
        self.op = op
        self.result_real = 0
        self.result_imag = 0
        self.calculate()
 
    def calculate(self):
        if self.op == '+':
            self.result_real = self.real1 + self.real2
            self.result_imag = self.imag1 + self.imag2
        elif self.op == '-':
            self.result_real = self.real1 - self.real2
            self.result_imag = self.imag1 - self.imag2
 
    def __str__(self):
        if self.result_real == 0 and self.result_imag == 0:
            return "0"
        elif self.result_real == 0:
            return f"{self.result_imag}i" if self.result_imag > 0 else f"-{abs(self.result_imag)}i"
        elif self.result_imag == 0:
            return str(self.result_real)
        else:
            sign = '+' if self.result_imag > 0 else ''
            return f"{self.result_real}{sign}{self.result_imag}i"
 
if __name__ == "__main__":
    real1, imag1, op, real2, imag2 = input().split()
    real1, imag1, real2, imag2 = map(int, [real1, imag1, real2, imag2])
    result = Complex(real1, imag1, real2, imag2, op)
    print(result)