Create describe-the-painting.py

Author: kamyu104

Python/describe-the-painting.py

@@ -0,0 +1,26 @@
+# Time:  O(nlogn)
+# Space: O(n)
+
+import collections
+
+
+class Solution(object):
+    def splitPainting(self, segments):
+        """
+        :type segments: List[List[int]]
+        :rtype: List[List[int]]
+        """
+        counts = collections.defaultdict(int)
+        for s, e, c in segments:
+            counts[s] += c
+            counts[e] -= c
+        points = sorted(x for x in counts.iteritems())
+
+        result = []
+        overlap = prev = 0
+        for curr, cnt in points:
+            if overlap:
+                result.append([prev, curr, overlap])
+            overlap += cnt
+            prev = curr
+        return result