feature: partition labels

arrays/ContainerWithMostWater.py

@@ -1,5 +1,6 @@
 from typing import List
 
+# TODO: two pointers
 
 # O(n) time || O(1) space
 def max_area(self, height: List[int]) -> int:

arrays/PartitionLabels.py

@@ -0,0 +1,16 @@
+from typing import List
+
+
+# O(n) time || O(n) space
+def partition_labels(self, s: str) -> List[int]:
+    last_occurrence = {c: i for i, c in enumerate(s)}
+    res = []
+    low = high = 0
+    for i, c in enumerate(s):
+        high = max(high, last_occurrence[c])
+
+        if i == high:
+            res.append(high - low + 1)
+            low = high + 1
+
+    return res

arrays/test/test_partition_labels.py

@@ -0,0 +1,15 @@
+import unittest
+
+from arrays.PartitionLabels import partition_labels
+
+
+class MyTestCase(unittest.TestCase):
+    def test_partition_labels(self):
+        self.assertEqual([9, 7, 8], partition_labels(self, s="ababcbacadefegdehijhklij"))
+
+    def test_partition_labels_1(self):
+        self.assertEqual([10], partition_labels(self, s="eccbbbbdec"))
+
+
+if __name__ == '__main__':
+    unittest.main()