Classes: Dealing with Complex Numbers HackerRank Solution In Python

Problem

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.

Input Format

One line of input: The real and imaginary part of a number separated by a space.

Output Format

For two complex numbers C and D, the output should be in the following sequence on separate lines:

• C + D
• C – D
• C * D
• C/D
• mod(C)
• mod(D)

For complex numbers with non-zero real(A) and complex part(B), the output should be in the following format:

A + Bi

Replace the plus symbol ( + ) with a minus symbol ( – ) when B < 0.

For complex numbers with a zero complex part i.e. real numbers, the output should be:

A + 0.00i

For complex numbers where the real part is zero and the complex part(B) is non-zero, the output should be:
0.00 + Bi

Sample Input

2 15 6

Sample Output

7.00+7.00i-3.00-5.00i4.00+17.00i0.26-0.11i2.24+0.00i7.81+0.00i

Concept

Python is a fully object-oriented language like C++, Java, etc. For reading about classes, refer here.

Methods with a double underscore before and after their name are considered as built-in methods. They are used by interpreters and are generally used in the implementation of overloaded operators or other built-in functionality.

__add__-> Can be overloaded for + operation

__sub__ -> Can be overloaded for - operation

__mul__ -> Can be overloaded for * operation

For more information on operator overloading in Python, refer here.

HackerRank Classes: Dealing with Complex Numbers Solution In Python

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

NOTE: The problem solved above, Classes: Dealing with Complex Numbers, was generated by HackerRank and the solution was brought by the admin of  CodingSolutions for educational purpose. Got any issues with the code? Ask your questions in the comment box and I shall attend to it.