HackerRank Python - Classes: Dealing with Complex Numbers


For this challenge, you are given two complex numbers, and you have to print the result of their addition, subtraction, multiplication, division and modulus operations.

The real and imaginary precision part should be correct up to two decimal places.



  1. import math

  2. 

  3. class Complex(object):

  4.     def __init__(self, real, imaginary):

  5.         self.real = real

  6.         self.imaginary = imaginary

  7. 

  8.     def __add__(self, no):

  9.         a = self.real

  10.         b = self.imaginary

  11.         c = no.real

  12.         d = no.imaginary

  13.         return Complex(a + c, b + d)

  14. 

  15.     def __sub__(self, no):

  16.         a = self.real

  17.         b = self.imaginary

  18.         c = no.real

  19.         d = no.imaginary

  20.         return Complex(a - c, b - d)

  21. 

  22.     def __mul__(self, no):

  23.         a = self.real

  24.         b = self.imaginary

  25.         c = no.real

  26.         d = no.imaginary

  27.         real_mult = (a * c) - (b * d)

  28.         imag_mult = (a * d) + (b * c)

  29.         return Complex(real_mult, imag_mult)

  30. 

  31.     def __truediv__(self, no):

  32.         a = self.real

  33.         b = self.imaginary

  34.         c = no.real

  35.         d = no.imaginary

  36.         real_numerator = a * c + b * d

  37.         imag_numerator = b * c - a * d

  38.         denom = c * c + d * d

  39.         real_div = real_numerator / denom

  40.         imag_div = imag_numerator / denom

  41.         return Complex(real_div, imag_div)

  42. 

  43.     def mod(self):

  44.         a = self.real

  45.         b = self.imaginary

  46.         return Complex(math.sqrt(a ** 2 + b ** 2), 0)

  47. 

  48.     def __str__(self):

  49.         if self.imaginary == 0:

  50.             result = '%.2f+0.00i' % (self.real)

  51.         elif self.real == 0:

  52.             if self.imaginary >= 0:

  53.                 result = '0.00+%.2fi' % (self.imaginary)

  54.             else:

  55.                 result = '0.00-%.2fi' % (abs(self.imaginary))

  56.         elif self.imaginary > 0:

  57.             result = '%.2f+%.2fi' % (self.real, self.imaginary)

  58.         else:

  59.             result = '%.2f-%.2fi' % (self.real, abs(self.imaginary))

  60.         return result

  61.     

  62. if __name__ == '__main__':

  63.     c = map(float, input().split())

  64.     d = map(float, input().split())

  65.     x = Complex(*c)

  66.     y = Complex(*d)

  67.     print(*map(str, [x+y, x-y, x*y, x/y, x.mod(), y.mod()]), sep='\n')
download android app