Update random-point-in-non-overlapping-rectangles.py

kamyu104 · web-flow · commit a70126b16a68 · 2018-07-29T02:45:53.000+08:00
diff --git a/Python/random-point-in-non-overlapping-rectangles.py b/Python/random-point-in-non-overlapping-rectangles.py
@@ -60,8 +60,7 @@ def pick(self):
         target = random.randint(0, self.__prefix_sum[-1]-1)
         left = bisect.bisect_right(self.__prefix_sum, target)
         rect = self.__rects[left]
-        width = rect[2]-rect[0]+1
-        height = rect[3]-rect[1]+1
+        width, height = rect[2]-rect[0]+1, rect[3]-rect[1]+1
         base = self.__prefix_sum[left]-width*height
         return [rect[0]+(target-base)%width, rect[1]+(target-base)//width]
 
