Update merge-intervals.py

kamyu104 · kamyu104 · commit 4406cef9bc9b · 2016-05-31T20:39:08.000+08:00
diff --git a/Python/merge-intervals.py b/Python/merge-intervals.py
@@ -17,21 +17,25 @@ def __init__(self, s=0, e=0):
     def __repr__(self):
         return "[{}, {}]".format(self.start, self.end)
 
-class Solution:
-    # @param intervals, a list of Interval
-    # @return a list of Interval
+
+class Solution(object):
     def merge(self, intervals):
+        """
+        :type intervals: List[Interval]
+        :rtype: List[Interval]
+        """
         if not intervals:
             return intervals
         intervals.sort(key = lambda x: x.start)
         result = [intervals[0]]
-        for i in range(1, len(intervals)):
+        for i in xrange(1, len(intervals)):
             prev, current = result[-1], intervals[i]
             if current.start <= prev.end: 
                 prev.end = max(prev.end, current.end)
             else:
                 result.append(current)
         return result
 
+
 if __name__ == "__main__":
     print Solution().merge([Interval(1, 3), Interval(2, 6), Interval(8, 10), Interval(15,18)])
