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')

728x90

저작자표시