forked from sknsht/HackerRank
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Solution.py
69 lines (59 loc) · 1.87 KB
/
Solution.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import math
class Complex(object):
def __init__(self, real, imaginary):
self.real = real
self.imaginary = imaginary
def __add__(self, no):
a = self.real
b = self.imaginary
c = no.real
d = no.imaginary
return Complex(a + c, b + d)
def __sub__(self, no):
a = self.real
b = self.imaginary
c = no.real
d = no.imaginary
return Complex(a - c, b - d)
def __mul__(self, no):
a = self.real
b = self.imaginary
c = no.real
d = no.imaginary
real_mult = (a * c) - (b * d)
imag_mult = (a * d) + (b * c)
return Complex(real_mult, imag_mult)
def __truediv__(self, no):
a = self.real
b = self.imaginary
c = no.real
d = no.imaginary
real_numerator = a * c + b * d
imag_numerator = b * c - a * d
denom = c * c + d * d
real_div = real_numerator / denom
imag_div = imag_numerator / denom
return Complex(real_div, imag_div)
def mod(self):
a = self.real
b = self.imaginary
return Complex(math.sqrt(a ** 2 + b ** 2), 0)
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 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')