forked from manishbisht/Competitive-Programming
-
Notifications
You must be signed in to change notification settings - Fork 1
/
75.Classes_ Dealing with Complex Numbers.py
58 lines (45 loc) · 1.73 KB
/
75.Classes_ Dealing with Complex Numbers.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
import math
class Complex(object):
def __init__(self, real, imaginary):
self.real=real
self.imaginary=imaginary
def __add__(self, no):
real=self.real+no.real
imaginary=self.imaginary+no.imaginary
return(Complex(real,imaginary))
def __sub__(self, no):
real=self.real-no.real
imaginary=self.imaginary-no.imaginary
return(Complex(real,imaginary))
def __mul__(self, no):
real=self.real*no.real-self.imaginary*no.imaginary
imaginary=self.real*no.imaginary+self.imaginary*no.real
return(Complex(real,imaginary))
def __truediv__(self, no):
x=float(no.real**2+no.imaginary**2)
y=self*Complex(no.real,-no.imaginary)
real=y.real/x
imaginary=y.imaginary/x
return(Complex(real,imaginary))
def mod(self):
real=math.sqrt(self.real**2+self.imaginary**2)
return(Complex(real,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')