Refactoring

Author: suyash248

Heap/HeapUtil.py

@@ -0,0 +1,52 @@
+from heapq import heappush, heappop, heapify
+from functools import total_ordering
+from enum import Enum
+from typing import TypeVar, List, Generic
+
+T = TypeVar('T')
+
+
+# why total_ordering: https://www.geeksforgeeks.org/python-functools-total_ordering/
+@total_ordering
+class __MinHeapElt__(Generic[T]):
+    def __init__(self, data: T):
+        self.data = data
+
+    def __lt__(self, other: T):
+        return self.data < other.data
+
+    def __eq__(self, other: T):
+        return self.data == other.data
+
+
+@total_ordering
+class __MaxHeapElt__(Generic[T]):
+    def __init__(self, data: T):
+        self.data = data
+
+    def __lt__(self, other: T):
+        return self.data > other.data
+
+    def __eq__(self, other: T):
+        return self.data == other.data
+
+
+class HeapType(Enum):
+    MIN = "MIN"
+    MAX = "MAX"
+
+
+class Heap(Generic[T]):
+    def __int__(self, heap_type: HeapType):
+        self._heap_: List[T] = []
+        self._heapType_: HeapType = heap_type
+        self._heapClass_ = __MinHeapElt__ if heap_type == HeapType.MIN else __MaxHeapElt__
+
+    def push(self, data: T):
+        heappush(self._heap_, self._heapClass_(data))
+
+    def pop(self) -> T:
+        return heappop(self._heap_)
+
+    def heapify(self):
+        heapify(self._heap_)

Heap/__init__.py

@@ -6,6 +6,7 @@ def __init__(self, data, next=None):
     def __str__(self):
         return "{} | next -> ".format(self.data, self.next.data if self.next else None)
 
+
 """
 Algorithm -
 
@@ -25,6 +26,8 @@ def __str__(self):
 
 References- https://www.geeksforgeeks.org/sum-of-two-linked-lists/
 """
+
+
 class Adder(object):
     def __init__(self, head1, head2):
         self.head1 = head1
@@ -97,7 +100,7 @@ def __propagate_carry__(self, head, sublist_head):
             self.result_head = new_node
 
             self.__sum__ += str(new_node.data)
-            #temp = temp.next
+            # temp = temp.next
 
     @classmethod
     def get_size(cls, head):
@@ -112,9 +115,9 @@ def get_size(cls, head):
     def print_list(cls, head):
         cur = head
         while cur is not None:
-            print str(cur.data) + "->",
+            print(str(cur.data) + "->")
             cur = cur.next
-        print None
+        print(None)
 
 
 def create_linked_list(data=None):
@@ -126,6 +129,7 @@ def create_linked_list(data=None):
         prev = cur
     return head
 
+
 # Input:
 #   First List: 5->6->3         // represents number 563
 #   Second List: 8->4->2        //  represents number 842
@@ -135,15 +139,15 @@ def create_linked_list(data=None):
     head1 = create_linked_list([2, 8, 9, 7])
     head2 = create_linked_list([1, 8])
 
-    print "List-1: ",
+    print("List-1: ")
     Adder.print_list(head1)
-    print "List-2: ",
+    print("List-2: ")
     Adder.print_list(head2)
 
     adder = Adder(head1, head2)
     result_sum, result_head = adder.add()
 
-    print "Result list: ",
+    print("Result list: ")
     Adder.print_list(result_head)
 
-    print "Total sum: {}".format(result_sum)
+    print("Total sum: {}".format(result_sum))

LinkedList/addLinkedLists.py

@@ -23,6 +23,8 @@ def __str__(self):
 
 References- https://www.geeksforgeeks.org/multiply-two-numbers-represented-linked-lists/
 """
+
+
 class Multiplier(object):
 
     def __init__(self, head1, head2):
@@ -57,9 +59,9 @@ def get_size(cls, head):
     def print_list(cls, head):
         cur = head
         while cur is not None:
-            print str(cur.data) + "->",
+            print(str(cur.data) + "->")
             cur = cur.next
-        print None
+        print(None)
 
 
 def create_linked_list(data=[]):
@@ -75,12 +77,12 @@ def create_linked_list(data=[]):
     head1 = create_linked_list([1, 0, 5])
     head2 = create_linked_list([1, 2])
 
-    print "List-1: ",
+    print("List-1: ")
     Multiplier.print_list(head1)
-    print "List-2: ",
+    print("List-2: ")
     Multiplier.print_list(head2)
 
     multiplier = Multiplier(head1, head2)
     result = multiplier.multiply()
 
-    print "Result: {}".format(result)
+    print("Result: {}".format(result))

LinkedList/multiplyLinkedLists.py

@@ -1,7 +1,7 @@
-from costExplorer.entity.customer import Customer
-from costExplorer.entity.plan import Plan
-from costExplorer.entity.product_subscription import ProductSubscription
-from costExplorer.service.cost_service import CostService
+from Misc.costExplorer.entity.customer import Customer
+from Misc.costExplorer.entity.plan import Plan
+from Misc.costExplorer.entity.product_subscription import ProductSubscription
+from Misc.costExplorer.service.cost_service import CostService
 
 if __name__ == '__main__':
     costService: CostService = CostService()

costExplorer/__init__.py Misc/costExplorer/__init__.py

@@ -1,4 +1,4 @@
-from costExplorer.entity.product_subscription import ProductSubscription
+from Misc.costExplorer.entity.product_subscription import ProductSubscription
 from typing import List
 
 class Customer(object):

costExplorer/driver.py Misc/costExplorer/driver.py

@@ -1,4 +1,4 @@
-from costExplorer.entity.plan import Plan
+from Misc.costExplorer.entity.plan import Plan
 
 
 class ProductSubscription(object):

costExplorer/entity/__init__.py Misc/costExplorer/entity/__init__.py

@@ -1,7 +1,7 @@
-from costExplorer.entity.customer import Customer
+from Misc.costExplorer.entity.customer import Customer
 from typing import List
 from datetime import date
-from costExplorer.entity.product_subscription import ProductSubscription
+from Misc.costExplorer.entity.product_subscription import ProductSubscription
 
 
 class CostService(object):

costExplorer/entity/customer.py Misc/costExplorer/entity/customer.py

@@ -45,8 +45,8 @@
         Vote(["Mykola"]),
     ]
 """
-from interview.main.vote import Vote
-from interview.main.voteUtility import VoteUtility
+from Misc.voting.main.vote import Vote
+from Misc.voting.main.voteUtility import VoteUtility
 
 if __name__ == '__main__':
     votes = [

costExplorer/entity/plan.py Misc/costExplorer/entity/plan.py

@@ -1,7 +1,5 @@
 from typing import List
 
-from interview.main.candidate import Candidate
-
 
 class Vote(object):
     def __init__(self, candidates: List[str]):

costExplorer/entity/product_subscription.py Misc/costExplorer/entity/product_subscription.py

@@ -1,4 +1,4 @@
-from interview.main.candidate import Candidate
+from Misc.voting.main.candidate import Candidate
 from typing import Dict, List
 from collections import defaultdict