Update count-number-of-rectangles-containing-each-point.py

Author: kamyu104

Python/count-number-of-rectangles-containing-each-point.py

@@ -18,10 +18,5 @@ def countRectangles(self, rectangles, points):
             buckets[y].append(x)
         for bucket in buckets:
             bucket.sort()
-        result = []
-        for x, y in points:
-            cnt = 0
-            for y in xrange(y, max_y+1):
-                cnt += len(buckets[y])-bisect.bisect_left(buckets[y], x)
-            result.append(cnt)
-        return result
+        return [sum(len(buckets[y])-bisect.bisect_left(buckets[y], x) for y in xrange(y, max_y+1))
+                for x, y in points]