Classes 1-2

Author: harshildarji

Classes/Class 2 Find the Torsional Angle.py

@@ -0,0 +1,51 @@
+# Class 2 - Find the Torsional Angle
+# https://www.hackerrank.com/challenges/class-2-find-the-torsional-angle/problem
+
+import math
+
+class vector:
+    def __init__(self, x=0, y=0, z=0):
+        self.x = x
+        self.y = y
+        self.z = z
+               
+    def __mul__(self, k):
+        return vector(self.x * k, self.y * k, self.z * k)
+    
+    def __add__(self, other):
+        s, o = self, other
+        return vector(s.x + o.x, s.y + o.y, s.z + o.z)
+   
+    def __sub__(self, other):
+        return self + other * -1
+        
+    # dot product
+    def __mod__(self, other):
+        s, o = self, other
+        return (s.x * o.x) + (s.y * o.y) + (s.z * o.z)
+    
+    # modulus
+    def __abs__(self):
+        return math.sqrt(self % self)
+    
+    # cross product
+    def __xor__(self, other):
+        s, o = self, other
+        x = (s.y * o.z) - (s.z * o.y)
+        y = (s.z * o.x) - (s.x * o.z)
+        z = (s.x * o.y) - (s.y * o.x)
+        return vector(x, y, z)
+    
+def read_vector():
+    return vector(*map(float, input().split()))
+    
+A = read_vector()
+B = read_vector()
+C = read_vector()
+D = read_vector()
+
+X = (B - A) ^ (C - B)
+Y = (C - B) ^ (D - C)
+
+phi = math.degrees(math.acos(X % Y / math.sqrt((X % X) * (Y % Y))))
+print('%.2f' % phi)

Classes/Classes Dealing with Complex Numbers.py

@@ -0,0 +1,56 @@
+# Classes: Dealing with Complex Numbers
+# https://www.hackerrank.com/challenges/class-1-dealing-with-complex-numbers/problem
+
+from math import sqrt
+
+class ComplexNumber(object):
+    def __init__(self, x=0, y=0):
+        self.real_p = x
+        self.imag_p = y
+
+    def __add__(self, other):
+        s = ComplexNumber()
+        s.real_p = self.real_p + other.real_p
+        s.imag_p = self.imag_p + other.imag_p
+        return s
+    
+    def __sub__(self, other):
+        s = ComplexNumber()
+        s.real_p = self.real_p - other.real_p
+        s.imag_p = self.imag_p - other.imag_p
+        return s
+
+    def __mul__(self, other):
+        s = ComplexNumber()
+        s.real_p = self.real_p*other.real_p - self.imag_p*other.imag_p
+        s.imag_p = self.real_p*other.imag_p + self.imag_p*other.real_p
+        return s
+
+    def __truediv__(self, other):
+        s = ComplexNumber()
+        s.real_p = (self.real_p*other.real_p + self.imag_p*other.imag_p) / (other.real_p**2 + other.imag_p**2)
+        s.imag_p = (self.imag_p*other.real_p - self.real_p*other.imag_p) / (other.real_p**2 + other.imag_p**2)
+        return s
+
+    def __abs__(self):
+        s = ComplexNumber()
+        s.real_p = sqrt(self.real_p**2 + self.imag_p**2)
+        return s
+
+    def __str__(self):
+        if self.imag_p >= 0:
+            s = "{0:.2f}+{1:.2f}i".format(self.real_p, self.imag_p)
+        else:
+            s = "{0:.2f}{1:.2f}i".format(self.real_p, self.imag_p)
+        return s
+
+x1, y1 = map(float, input().split())
+x2, y2 = map(float, input().split())
+z1 = ComplexNumber(x1, y1)
+z2 = ComplexNumber(x2, y2)
+print(z1+z2)
+print(z1-z2)
+print(z1*z2)
+print(z1/z2)
+print(abs(z1))
+print(abs(z2))