solve : insert interval

Author: SamTheKorean

insert-interval/samthekorean.py

@@ -0,0 +1,26 @@
+# TC : O(n + m)
+# SC : O(n + m)
+# n is a size of intervals and m is a size of newInterval
+class Solution:
+    def insert(
+        self, intervals: List[List[int]], newInterval: List[int]
+    ) -> List[List[int]]:
+        result = []
+        i = 0
+
+        while i < len(intervals) and intervals[i][1] < newInterval[0]:
+            result.append(intervals[i])
+            i += 1
+
+        while i < len(intervals) and intervals[i][0] <= newInterval[1]:
+            newInterval[0] = min(newInterval[0], intervals[i][0])
+            newInterval[1] = max(newInterval[1], intervals[i][1])
+            i += 1
+
+        result.append(newInterval)
+
+        while i < len(intervals):
+            result.append(intervals[i])
+            i += 1
+
+        return result