910. Smallest Range II


Problem Description

In this problem, you are given an integer array called nums and an integer k. Your task is to modify each element in the array by either adding k to it or subtracting k from it. After modifying each element in this manner, the score of the array is defined as the difference between the maximum and minimum elements in the modified array. The objective is to determine the minimum score that can be achieved by any combination of adding or subtracting k to/from the elements of nums.

Here is an example to illustrate:

Suppose nums = [1, 3, 6] and k = 3. You could transform nums as follows:

  • Adding k to the first element: [1 + 3, 3, 6] = [4, 3, 6]
  • Subtracting k from the second element: [4, 3 - 3, 6] = [4, 0, 6]
  • Subtracting k from the third element: [4, 0, 6 - 3] = [4, 0, 3]

The score is then max(4, 0, 3) - min(4, 0, 3) = 4 - 0 = 4.

The problem is asking you to find the minimum score possible, that is, the smallest difference between the highest and lowest number in the array after each element has been increased or decreased by k.

Intuition

The intuition behind the solution involves recognizing that sorting the array can help in minimizing the score. By sorting, you can ensure that the operation that you perform (addition or subtraction) will not increase the range unnecessarily.

Here's why sorting helps:

  • After sorting nums, the smallest and largest elements are nums[0] and nums[-1], respectively. The initial score is nums[-1] - nums[0].
  • Consider a pivot point at index i in the sorted array. Everything to the left of i could be increased by k, and everything to the right of i (including element at i) could be decreased by k. This creates two "blocks" within the array: one with increased values, and one with decreased values.
  • The new minimum possible value is the minimum of nums[0] + k (left-most element of the increased block) and nums[i] - k (left-most element of the decreased block).
  • The new maximum possible value is the maximum of nums[i - 1] + k (right-most element of the increased block) and nums[-1] - k (right-most element of the decreased block).
  • By iterating through all possible pivot points (from 1 to len(nums) - 1), you can find the minimum score by comparing this with the previously calculated minimum ans.

The key observation here is that by sorting and choosing an appropriate pivot, one can minimize the difference between the maximum and minimum values affected by the addition or subtraction of k, leading directly to a solution that iterates through possible pivot points to find the minimum score.

Learn more about Greedy, Math and Sorting patterns.

Solution Approach

The solution follows a simple yet effective approach leveraging sorting and iteration.

Algorithm:

  1. Sort the nums array. This will help us easily identify the smallest and largest elements and ensures that we do not increase the range unnecessarily.

  2. Initialize the variable ans with the initial score, which is the difference between the last element and the first element of the sorted array (nums[-1] - nums[0]).

  3. Iterate through the array starting from index 1 up to len(nums) - 1. The reason we start at 1 is because we are considering the pivot point where the array is divided into two parts: one that will get the addition of k and the other the subtraction. There is no point in considering index 0 for this pivot as the sorted array's first element cannot be the start of the subtraction section.

  4. For each index i, calculate the minimum and maximum values as follows:

    • Calculate the new minimum value mi as the minimum between nums[0] + k and nums[i] - k. The nums[0] + k represents the smallest possible value after the increment and nums[i] - k represents the smallest value for the section of the array where k is subtracted.

    • Calculate the new maximum value mx as the maximum between nums[i - 1] + k and nums[-1] - k. The nums[i - 1] + k is the largest value of the incremented section, and nums[-1] - k is the largest value for the decremented section.

  5. Update the score (ans) to be the minimum value between the current ans and the difference mx - mi. This ensures that with each iteration, we are considering the lowest possible range after applying our operation at each pivot.

  6. Once the loop completes, ans contains the minimum score achievable after performing the add or subtract operation at each index, based on the given k.

Data Structures:

  • The sorted array itself is the primary data structure used. No additional data structures are required.

Patterns:

  • The use of sorting to establish a predictable order of elements, hence allowing for a methodical approach to finding the solution.
  • Iteration to explore possible optimal solutions by checking pivot points in the array.
  • Decision-making to update the candidate solution (ans) based on comparisons calculated within the loop.

By following these steps, the algorithm ensures that we calculate the minimum range after adding or subtracting k from each element in the most efficient manner. It elegantly handles the problem by transforming it into a series of operations where we only need to look at the edges of the two blocks created by the pivot point.

Ready to land your dream job?

Unlock your dream job with a 2-minute evaluator for a personalized learning plan!

Start Evaluator

Example Walkthrough

Let's go through the solution approach with a small example to better illustrate the algorithm.

Suppose we have the following nums array and integer k:

nums = [4, 7, 1]
k = 5

According to the problem, we have to either add k to or subtract k from each element in the array to achieve the smallest possible score (the difference between the maximum and minimum elements of the array). Let's follow the solution approach:

  1. Sort the array:

    nums.sort() -> nums = [1, 4, 7]
  2. Initialize the score (ans):

    ans = nums[-1] - nums[0] -> ans = 7 - 1 -> ans = 6
  3. Iterate through the array starting from index 1:

    For each index i, we will calculate the potential new minimum (mi) and maximum (mx) values after either adding or subtracting k:

    • At index i = 1 (element 4):

      mi = min(nums[0] + k, nums[i] - k) -> mi = min(1 + 5, 4 - 5) -> mi = min(6, -1) -> mi = -1
      mx = max(nums[i - 1] + k, nums[-1] - k) -> mx = max(1 + 5, 7 - 5) -> mx = max(6, 2) -> mx = 6

      Now, we update the ans if mx - mi is smaller than the current ans:

      ans = min(ans, mx - mi) -> ans = min(6, 6 - (-1)) -> ans = min(6, 7) -> ans = 6

      (No change in ans as the score is still 6 which is not better than the previous score)

    • At index i = 2 (element 7):

      mi = min(nums[0] + k, nums[i] - k) -> mi = min(1 + 5, 7 - 5) -> mi = min(6, 2) -> mi = 2
      mx = max(nums[i - 1] + k, nums[-1] - k) -> mx = max(4 + 5, 7 - 5) -> mx = max(9, 2) -> mx = 9

      Now, update the ans:

      ans = min(ans, mx - mi) -> ans = min(6, 9 - 2) -> ans = min(6, 7) -> ans = 6

      Again, the score is 7, which does not improve the ans.

  4. Complete the loop: The algorithm has finished checking all possible pivot points in the array.

  5. Result: The ans calculated is 6, which means:

    The minimum score achievable after adding or subtracting `k` from each element in nums is 6.

In this example, the modifications to the elements are unnecessary, as the initial score is already minimal. The algorithm efficiently identifies this by analyzing the potential effects of adding and subtracting k at each pivot point in the sorted array.

Solution Implementation

1from typing import List
2
3class Solution:
4    def smallestRangeII(self, nums: List[int], k: int) -> int:
5        # First, sort the numbers to organize them in ascending order.
6        nums.sort()
7
8        # The initial range would be the max value minus the min value.
9        smallest_range = nums[-1] - nums[0]
10
11        # Loop through the sorted numbers, starting from the second element,
12        # to find the minimum possible range.
13        for i in range(1, len(nums)):
14            # Calculate the possible minimum by adding k to the smallest value
15            # and subtracting k from the current value.
16            possible_min = min(nums[0] + k, nums[i] - k)
17
18            # Calculate the possible maximum by adding k to the previous value
19            # and subtracting k from the maximum value.
20            possible_max = max(nums[i - 1] + k, nums[-1] - k)
21
22            # Update the smallest range if a smaller one is found.
23            smallest_range = min(smallest_range, possible_max - possible_min)
24
25        # Finally, return the smallest range after considering all elements.
26        return smallest_range
27
1class Solution {
2    public int smallestRangeII(int[] nums, int k) {
3        // First, sort the input array to deal with numbers in a sorted order.
4        Arrays.sort(nums);
5        // Get the length of the array
6        int n = nums.length;
7        // Calculate the initial range from the first and last element of the sorted array.
8        int minRange = nums[n - 1] - nums[0];
9      
10        // Iterate through the array starting from the second element
11        for (int i = 1; i < n; ++i) {
12            // Calculate the minimum possible value after adding or subtracting k to the current or first element
13            int currentMin = Math.min(nums[0] + k, nums[i] - k);
14            // Calculate the maximum possible value after adding or subtracting k to the previous or last element
15            int currentMax = Math.max(nums[i - 1] + k, nums[n - 1] - k);
16            // Find the smallest range by comparing the current computed range and the previously stored minimum range
17            minRange = Math.min(minRange, currentMax - currentMin);
18        }
19        // Return the minimum range found
20        return minRange;
21    }
22}
23
1class Solution {
2public:
3    // Defines the function which will find the smallest range after modification
4    int smallestRangeII(vector<int>& nums, int k) {
5        // Initially, sort the array in non-decreasing order
6        sort(nums.begin(), nums.end());
7      
8        int numsSize = nums.size(); // Store the size of the nums vector
9        // Compute the initial range between the largest and smallest numbers
10        int minRange = nums[numsSize - 1] - nums[0];
11      
12        // Iterate through the sorted numbers starting from the second element
13        for (int i = 1; i < numsSize; ++i) {
14            // Determine the new minimum by comparing the increased smallest number
15            // and the decreased current number
16            int newMin = min(nums[0] + k, nums[i] - k);
17            // Determine the new maximum by comparing the increased previous number
18            // and the decreased largest number
19            int newMax = max(nums[i - 1] + k, nums[numsSize - 1] - k);
20            // Update the minRange with the minimum of the current and new ranges
21            minRange = min(minRange, newMax - newMin);
22        }
23      
24        // Return the smallest range found
25        return minRange;
26    }
27};
28
1// Array `nums` holds the numbers and `k` is the allowed modification range
2let nums: number[];
3let k: number;
4
5// Sorts the array in non-decreasing order
6const sortArray = (a: number, b: number) => a - b;
7
8// Finds the smallest range after modification
9function smallestRangeII(nums: number[], k: number): number {
10    // Sort the array in non-decreasing order
11    nums.sort(sortArray);
12
13    // Store the size of the nums array
14    const numsSize: number = nums.length;
15  
16    // Compute the initial range between the largest and smallest numbers
17    let minRange: number = nums[numsSize - 1] - nums[0];
18  
19    // Iterate through the sorted numbers starting from the second element
20    for (let i = 1; i < numsSize; ++i) {
21        // Determine the new minimum by comparing the increased smallest number and the decreased current number
22        const newMin: number = Math.min(nums[0] + k, nums[i] - k);
23      
24        // Determine the new maximum by comparing the increased previous number and the decreased largest number
25        const newMax: number = Math.max(nums[i - 1] + k, nums[numsSize - 1] - k);
26      
27        // Update the minRange with the minimum of the current and new ranges
28        minRange = Math.min(minRange, newMax - newMin);
29    }
30  
31    // Return the smallest range found
32    return minRange;
33}
34
35// Example usage:
36// nums = [1, 3, 6];
37// k = 3;
38// console.log(smallestRangeII(nums, k)); // Output would be the result of the smallestRangeII function
39

Time and Space Complexity

Time Complexity

The time complexity of the provided code depends on the sorting algorithm and the for loop.

  1. nums.sort() -> The sort operation typically has a time complexity of O(n log n) where n is the number of elements in the list nums.

  2. The for loop -> Iterates through the sorted list once, accounting for a time complexity of O(n).

Combining both, the time complexity remains dominated by the sorting step, and therefore, the overall time complexity is O(n log n).

Space Complexity

The space complexity of the provided code is mainly due to the sorted list.

  1. nums.sort() -> The sort operation in Python is usually done in-place, which means the space complexity is O(1).

  2. No additional data structures are used that are dependent on the number of elements in the list, and therefore, no extra space is utilized that is proportional to the input size.

Thus, the space complexity of the code is O(1).

Learn more about how to find time and space complexity quickly using problem constraints.


Discover Your Strengths and Weaknesses: Take Our 2-Minute Quiz to Tailor Your Study Plan:
Question 1 out of 10

What's the output of running the following function using input [30, 20, 10, 100, 33, 12]?

1def fun(arr: List[int]) -> List[int]:
2    import heapq
3    heapq.heapify(arr)
4    res = []
5    for i in range(3):
6        res.append(heapq.heappop(arr))
7    return res
8
1public static int[] fun(int[] arr) {
2    int[] res = new int[3];
3    PriorityQueue<Integer> heap = new PriorityQueue<>();
4    for (int i = 0; i < arr.length; i++) {
5        heap.add(arr[i]);
6    }
7    for (int i = 0; i < 3; i++) {
8        res[i] = heap.poll();
9    }
10    return res;
11}
12
1class HeapItem {
2    constructor(item, priority = item) {
3        this.item = item;
4        this.priority = priority;
5    }
6}
7
8class MinHeap {
9    constructor() {
10        this.heap = [];
11    }
12
13    push(node) {
14        // insert the new node at the end of the heap array
15        this.heap.push(node);
16        // find the correct position for the new node
17        this.bubble_up();
18    }
19
20    bubble_up() {
21        let index = this.heap.length - 1;
22
23        while (index > 0) {
24            const element = this.heap[index];
25            const parentIndex = Math.floor((index - 1) / 2);
26            const parent = this.heap[parentIndex];
27
28            if (parent.priority <= element.priority) break;
29            // if the parent is bigger than the child then swap the parent and child
30            this.heap[index] = parent;
31            this.heap[parentIndex] = element;
32            index = parentIndex;
33        }
34    }
35
36    pop() {
37        const min = this.heap[0];
38        this.heap[0] = this.heap[this.size() - 1];
39        this.heap.pop();
40        this.bubble_down();
41        return min;
42    }
43
44    bubble_down() {
45        let index = 0;
46        let min = index;
47        const n = this.heap.length;
48
49        while (index < n) {
50            const left = 2 * index + 1;
51            const right = left + 1;
52
53            if (left < n && this.heap[left].priority < this.heap[min].priority) {
54                min = left;
55            }
56            if (right < n && this.heap[right].priority < this.heap[min].priority) {
57                min = right;
58            }
59            if (min === index) break;
60            [this.heap[min], this.heap[index]] = [this.heap[index], this.heap[min]];
61            index = min;
62        }
63    }
64
65    peek() {
66        return this.heap[0];
67    }
68
69    size() {
70        return this.heap.length;
71    }
72}
73
74function fun(arr) {
75    const heap = new MinHeap();
76    for (const x of arr) {
77        heap.push(new HeapItem(x));
78    }
79    const res = [];
80    for (let i = 0; i < 3; i++) {
81        res.push(heap.pop().item);
82    }
83    return res;
84}
85

Recommended Readings

Want a Structured Path to Master System Design Too? Don’t Miss This!


Load More