Time: 302 ms (13.38%), Space: 101.7 MB (10.24%) - LeetHub

Author: MishkatIT

0945-minimum-increment-to-make-array-unique/0945-minimum-increment-to-make-array-unique.cpp

@@ -0,0 +1,25 @@
+class Solution {
+public:
+    int minIncrementForUnique(vector<int>& nums) {
+        map<int, int> mp;
+        for (auto& i : nums) {
+            mp[i]++;
+        }
+        long long sum = -1;
+        long long ans = 0;
+        for (auto& i : mp) {
+            if (sum < i.first) {
+                long long x = i.second - 1;
+                ans += x * (x + 1) / 2;
+                sum = i.first + x;
+            } else {
+                long long x = i.second - 1;
+                ans += (sum - i.first + 1) * i.second;
+                ans += x * (x + 1) / 2;
+                sum += i.second;
+            }
+
+        }
+        return ans;
+    }
+};