Update exam-room.py

Author: kamyu104

Python/exam-room.py

@@ -13,7 +13,7 @@ def __init__(self, N):
         """
         self.__num = N
         self.__seats = {-1: [-1, self.__num], self.__num: [-1, self.__num]}
-        self.__max_heap = [(-self.distance((-1, self.__num)), -1, self.__num)]
+        self.__max_heap = [(-self.__distance((-1, self.__num)), -1, self.__num)]
 
     def seat(self):
         """
@@ -30,8 +30,8 @@ def seat(self):
               else self.__num-1 if right == self.__num \
               else (left+right) // 2
         self.__seats[mid] =  [left, right]
-        heapq.heappush(self.__max_heap, (-self.distance((left, mid)), left, mid))
-        heapq.heappush(self.__max_heap, (-self.distance((mid, right)), mid, right))
+        heapq.heappush(self.__max_heap, (-self.__distance((left, mid)), left, mid))
+        heapq.heappush(self.__max_heap, (-self.__distance((mid, right)), mid, right))
         self.__seats[left][1] = mid
         self.__seats[right][0] = mid
         return mid
@@ -45,9 +45,9 @@ def leave(self, p):
         self.__seats.pop(p)
         self.__seats[left][1] = right
         self.__seats[right][0] = left
-        heapq.heappush(self.__max_heap, (-self.distance((left, right)), left, right))
+        heapq.heappush(self.__max_heap, (-self.__distance((left, right)), left, right))
 
-    def distance(self, segment):
+    def __distance(self, segment):
         return segment[1]-segment[0]-1 if segment[0] == -1 or segment[1] == self.__num \
                else (segment[1]-segment[0]) // 2