945 accepted. 72ms faster than 96.55%

Author: aQuaYi

Algorithms/0945.minimum-increment-to-make-array-unique/minimum-increment-to-make-array-unique.go

@@ -1,24 +1,19 @@
 package problem0945
 
-func minIncrementForUnique(A []int) int {
-	size := len(A)
-	if size == 0 {
-		return 0
-	}
+import "sort"
 
-	rec := [80000]bool{}
+func minIncrementForUnique(A []int) int {
 
-	rec[A[0]] = true
+	sort.Ints(A)
 
+	size := len(A)
 	res := 0
-
 	for i := 1; i < size; i++ {
-		n := A[i]
-		for rec[n] {
-			n++
-			res++
+		d := A[i-1] - A[i] + 1
+		if d > 0 {
+			res += d
+			A[i] += d
 		}
-		rec[n] = true
 	}
 
 	return res