Create smallest-range-ii.cpp

rui-yan · Sep 23, 2018 · a61b88f · a61b88f
1 parent 089b7b5
commit a61b88f
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/C++/smallest-range-ii.cpp b/C++/smallest-range-ii.cpp
@@ -0,0 +1,16 @@
+// Time:  O(nlogn)
+// Space: O(1)
+
+class Solution {
+public:
+    int smallestRangeII(vector<int>& A, int K) {
+        sort(A.begin(), A.end());
+        int result = A.back() - A.front();
+        for (int i = 0; i < A.size() - 1; ++i) {
+            result = min(result,
+                         max(A.back() - K, A[i] + K) - 
+                         min(A.front() + K, A[i + 1] - K));
+        }
+        return result;
+    }
+};