Create minimum-increment-to-make-array-unique.py

Author: kamyu104

Python/minimum-increment-to-make-array-unique.py

@@ -0,0 +1,21 @@
+# Time:  O(nlogn)
+# Space: O(n)
+
+class Solution(object):
+    def minIncrementForUnique(self, A):
+        """
+        :type A: List[int]
+        :rtype: int
+        """
+        A.sort()
+        A.append(float("inf"))
+        result, duplicate = 0, 0
+        for i in xrange(1, len(A)):
+            if A[i-1] == A[i]:
+                duplicate += 1
+                result -= A[i]
+            else:
+                move = min(duplicate, A[i]-A[i-1]-1)
+                duplicate -= move
+                result += move*A[i-1] + move*(move+1)//2
+        return result