Practice>Python>Classes>Classes>Dealing with Complex Numbers

hackerrank.com

Dealing with Complex Numbers

Problem link : https://www.hackerrank.com/challenges/class-1-dealing-with-complex-numbers/problem

Difficultiy : Medium

Tip

• This problem is to check that you know how to use complex() from cmath
• we map each arithmetic operation is each method to complex(), it's all you should do

Solution

class Complex(complex): def __add__(self, o): return Complex(complex.__add__(self, o)) def __sub__(self, o): return Complex(complex.__sub__(self, o)) def __mul__(self, o): return Complex(complex.__mul__(self, o)) def __truediv__(self, o): return Complex(complex.__truediv__(self, o)) def mod(self): return Complex(complex.__abs__(self)) def __str__(self): if self.imag == 0: result = "%.2f+0.00i" % (self.real) elif self.real == 0: if self.imag >= 0: result = "0.00+%.2fi" % (self.imag) else: result = "0.00-%.2fi" % (abs(self.imag)) elif self.imag > 0: result = "%.2f+%.2fi" % (self.real, self.imag) else: result = "%.2f-%.2fi" % (self.real, abs(self.imag)) return result if __name__ == '__main__': c = map(float, input().split()) d = map(float, input().split()) x = Complex(*c) y = Complex(*d) print(*map(str, [x+y, x-y, x*y, x/y, x.mod(), y.mod()]), sep='\n')