Create duplicate-zeros.py

Author: kamyu104

Python/duplicate-zeros.py

@@ -0,0 +1,21 @@
+# Time:  O(n)
+# Space: O(1)
+
+class Solution(object):
+    def duplicateZeros(self, arr):
+        """
+        :type arr: List[int]
+        :rtype: None Do not return anything, modify arr in-place instead.
+        """
+        shift, i = 0, 0
+        while i+shift < len(arr):
+            shift += int(arr[i] == 0)
+            i += 1
+        i -= 1
+        while shift:
+            if i+shift < len(arr):
+                arr[i+shift] = arr[i]
+            if arr[i] == 0:
+                shift -= 1
+                arr[i+shift] = arr[i]
+            i -= 1