624. Maximum Distance in Arrays - Explanation

Description

You are given m arrays, where each array is sorted in ascending order.

You can pick up two integers from two different arrays (each array picks one) and calculate the distance. We define the distance between two integers a and b to be their absolute difference |a - b|.

Return the maximum distance.

Example 1:

Input: arrays = [[1,2,3],[4,5],[1,2,3]]

Output: 4

Explanation: One way to reach the maximum distance 4, is to pick 1 in the first or third array and pick 5 in the second array.

Example 2:

Input: arrays = [[1],[1]]

Output:  0

Constraints:

m == arrays.length
2 <= m <= 10^5
1 <= arrays[i].length <= 500
-10^4 <= arrays[i][j] <= 10^4
arrays[i] is sorted in ascending order.
There will be at most 10^5 integers in all the arrays.

Topics

Array Greedy

Company Tags

Please upgrade to NeetCode Pro to view company tags.

Prerequisites

Before attempting this problem, you should be comfortable with:

Sorted Array Properties - Understanding that the minimum is at the start and maximum at the end of a sorted array
Tracking Global Extremes - Maintaining running minimum and maximum values across multiple data sources
Greedy Approach - Making locally optimal choices to build towards the global solution

1. Brute Force

Intuition

We need to find the maximum absolute difference between elements from two different arrays. The naive approach is to compare every element from every array with every element from every other array. This guarantees we find the answer but is slow due to the nested iteration over all elements.

Algorithm

Use four nested loops: the outer two select a pair of different arrays, and the inner two select elements from each array.
For each pair of elements from different arrays, compute the absolute difference.
Track the maximum difference found.
Return the result.

class Solution:
    def maxDistance(self, arrays: list[list[int]]) -> int:
        res = 0
        n = len(arrays)
        for i in range(n - 1):
            for j in range(len(arrays[i])):
                for k in range(i + 1, n):
                    for l in range(len(arrays[k])):
                        res = max(res, abs(arrays[i][j] - arrays[k][l]))
        return res

class Solution {
    public int maxDistance(List<List<Integer>> arrays) {
        int res = 0;
        int n = arrays.size();
        for (int i = 0; i < n - 1; i++) {
            for (int j = 0; j < arrays.get(i).size(); j++) {
                for (int k = i + 1; k < n; k++) {
                    for (int l = 0; l < arrays.get(k).size(); l++) {
                        res = Math.max(res, Math.abs(arrays.get(i).get(j) - arrays.get(k).get(l)));
                    }
                }
            }
        }
        return res;
    }
}

class Solution {
public:
    int maxDistance(std::vector<std::vector<int>>& arrays) {
        int res = 0;
        int n = arrays.size();
        for (int i = 0; i < n - 1; i++) {
            for (int j = 0; j < arrays[i].size(); j++) {
                for (int k = i + 1; k < n; k++) {
                    for (int l = 0; l < arrays[k].size(); l++) {
                        res = std::max(res, std::abs(arrays[i][j] - arrays[k][l]));
                    }
                }
            }
        }
        return res;
    }
};

class Solution {
    /**
     * @param {number[][]} arrays
     * @return {number}
     */
    maxDistance(arrays) {
        let res = 0;
        const n = arrays.length;
        for (let i = 0; i < n - 1; i++) {
            for (let j = 0; j < arrays[i].length; j++) {
                for (let k = i + 1; k < n; k++) {
                    for (let l = 0; l < arrays[k].length; l++) {
                        res = Math.max(res, Math.abs(arrays[i][j] - arrays[k][l]));
                    }
                }
            }
        }
        return res;
    }
}

public class Solution {
    public int MaxDistance(IList<IList<int>> arrays) {
        int res = 0;
        int n = arrays.Count;
        for (int i = 0; i < n - 1; i++) {
            for (int j = 0; j < arrays[i].Count; j++) {
                for (int k = i + 1; k < n; k++) {
                    for (int l = 0; l < arrays[k].Count; l++) {
                        res = Math.Max(res, Math.Abs(arrays[i][j] - arrays[k][l]));
                    }
                }
            }
        }
        return res;
    }
}

class Solution {
    fun maxDistance(arrays: List<List<Int>>): Int {
        var res = 0
        val n = arrays.size
        for (i in 0 until n - 1) {
            for (j in arrays[i].indices) {
                for (k in i + 1 until n) {
                    for (l in arrays[k].indices) {
                        res = maxOf(res, kotlin.math.abs(arrays[i][j] - arrays[k][l]))
                    }
                }
            }
        }
        return res
    }
}

class Solution {
    func maxDistance(_ arrays: [[Int]]) -> Int {
        var res = 0
        let n = arrays.count
        for i in 0..<(n - 1) {
            for j in 0..<arrays[i].count {
                for k in (i + 1)..<n {
                    for l in 0..<arrays[k].count {
                        res = max(res, abs(arrays[i][j] - arrays[k][l]))
                    }
                }
            }
        }
        return res
    }
}

Time & Space Complexity

Time complexity: $O((n * x)^2)$
Space complexity: $O(1)$ constant space used

Where $n$ refers to the number of arrays in $arrays$ and $x$ refers to the average number of elements in each array in $arrays$ .

2. Better Brute Force

Intuition

Since each array is sorted, the minimum is always at the start and the maximum is always at the end. This means we only need to consider the first and last elements of each array. For any pair of arrays, the maximum distance is either |min1 - max2| or |min2 - max1|. This reduces the inner loops to constant time per pair.

Algorithm

For each pair of arrays (i, j) where i < j:
- Compute |array1[0] - array2[last]| and |array2[0] - array1[last]|.
- Update the maximum distance.
Return the maximum distance found.

class Solution:
    def maxDistance(self, arrays: list[list[int]]) -> int:
        res = 0
        n = len(arrays)
        for i in range(n - 1):
            for j in range(i + 1, n):
                array1 = arrays[i]
                array2 = arrays[j]
                res = max(res, abs(array1[0] - array2[-1]))
                res = max(res, abs(array2[0] - array1[-1]))
        return res

class Solution {
    public int maxDistance(List<List<Integer>> arrays) {
        List<Integer> array1, array2;
        int res = 0;
        int n = arrays.size();
        for (int i = 0; i < n - 1; i++) {
            for (int j = i + 1; j < n; j++) {
                array1 = arrays.get(i);
                array2 = arrays.get(j);
                res = Math.max(res, Math.abs(array1.get(0) - array2.get(array2.size() - 1)));
                res = Math.max(res, Math.abs(array2.get(0) - array1.get(array1.size() - 1)));
            }
        }
        return res;
    }
}

class Solution {
public:
    int maxDistance(std::vector<std::vector<int>>& arrays) {
        std::vector<int> array1, array2;
        int res = 0;
        int n = arrays.size();
        for (int i = 0; i < n - 1; i++) {
            for (int j = i + 1; j < n; j++) {
                array1 = arrays[i];
                array2 = arrays[j];
                res = std::max(res, std::abs(array1[0] - array2[array2.size() - 1]));
                res = std::max(res, std::abs(array2[0] - array1[array1.size() - 1]));
            }
        }
        return res;
    }
};

class Solution {
    /**
     * @param {number[][]} arrays
     * @return {number}
     */
    maxDistance(arrays) {
        let array1, array2;
        let res = 0;
        const n = arrays.length;
        for (let i = 0; i < n - 1; i++) {
            for (let j = i + 1; j < n; j++) {
                array1 = arrays[i];
                array2 = arrays[j];
                res = Math.max(res, Math.abs(array1[0] - array2[array2.length - 1]));
                res = Math.max(res, Math.abs(array2[0] - array1[array1.length - 1]));
            }
        }
        return res;
    }
}

public class Solution {
    public int MaxDistance(IList<IList<int>> arrays) {
        IList<int> array1, array2;
        int res = 0;
        int n = arrays.Count;
        for (int i = 0; i < n - 1; i++) {
            for (int j = i + 1; j < n; j++) {
                array1 = arrays[i];
                array2 = arrays[j];
                res = Math.Max(res, Math.Abs(array1[0] - array2[array2.Count - 1]));
                res = Math.Max(res, Math.Abs(array2[0] - array1[array1.Count - 1]));
            }
        }
        return res;
    }
}

func maxDistance(arrays [][]int) int {
    var array1, array2 []int
    res := 0
    n := len(arrays)
    for i := 0; i < n-1; i++ {
        for j := i + 1; j < n; j++ {
            array1 = arrays[i]
            array2 = arrays[j]
            res = max(res, abs(array1[0] - array2[len(array2)-1]))
            res = max(res, abs(array2[0] - array1[len(array1)-1]))
        }
    }
    return res
}

func abs(x int) int {
    if x < 0 {
        return -x
    }
    return x
}

func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}

class Solution {
    fun maxDistance(arrays: List<List<Int>>): Int {
        var res = 0
        val n = arrays.size
        for (i in 0 until n - 1) {
            for (j in i + 1 until n) {
                val array1 = arrays[i]
                val array2 = arrays[j]
                res = maxOf(res, kotlin.math.abs(array1[0] - array2[array2.size - 1]))
                res = maxOf(res, kotlin.math.abs(array2[0] - array1[array1.size - 1]))
            }
        }
        return res
    }
}

class Solution {
    func maxDistance(_ arrays: [[Int]]) -> Int {
        var res = 0
        let n = arrays.count
        for i in 0..<(n - 1) {
            for j in (i + 1)..<n {
                let array1 = arrays[i]
                let array2 = arrays[j]
                res = max(res, abs(array1[0] - array2[array2.count - 1]))
                res = max(res, abs(array2[0] - array1[array1.count - 1]))
            }
        }
        return res
    }
}

Time & Space Complexity

Time complexity: $O(n^2)$
Space complexity: $O(1)$ constant space used

Where $n$ is the number of arrays in $arrays$

3. Single Scan

Intuition

We can do better by making a single pass. As we scan through the arrays, we maintain the global minimum and maximum seen so far. For each new array, the best distance involving this array is either current_max - global_min or global_max - current_min. After checking, we update our global min and max to include the current array's values.

The key insight is that we compare the current array against all previous arrays implicitly through the running min and max. This guarantees the two elements come from different arrays.

Algorithm

Initialize min_val and max_val from the first array.
For each subsequent array:
- Compute the distance using current_last - min_val and max_val - current_first.
- Update the result with the maximum of these distances.
- Update min_val and max_val to include the current array's first and last elements.
Return the result.

class Solution:
    def maxDistance(self, arrays: list[list[int]]) -> int:
        res = 0
        n = len(arrays[0])
        min_val = arrays[0][0]
        max_val = arrays[0][-1]
        for i in range(1, len(arrays)):
            n = len(arrays[i])
            res = max(res, max(abs(arrays[i][n - 1] - min_val), 
                               abs(max_val - arrays[i][0])))
            min_val = min(min_val, arrays[i][0])
            max_val = max(max_val, arrays[i][n - 1])
        return res

class Solution {
    public int maxDistance(List<List<Integer>> arrays) {
        int res = 0;
        int n = arrays.get(0).size();
        int min_val = arrays.get(0).get(0);
        int max_val = arrays.get(0).get(arrays.get(0).size() - 1);
        for (int i = 1; i < arrays.size(); i++) {
            n = arrays.get(i).size();
            res = Math.max(res, Math.max(Math.abs(arrays.get(i).get(n - 1) - min_val), 
                                         Math.abs(max_val - arrays.get(i).get(0))));
            min_val = Math.min(min_val, arrays.get(i).get(0));
            max_val = Math.max(max_val, arrays.get(i).get(n - 1));
        }
        return res;
    }
}

class Solution {
public:
    int maxDistance(std::vector<std::vector<int>>& arrays) {
        int res = 0;
        int n = arrays[0].size();
        int min_val = arrays[0][0];
        int max_val = arrays[0][arrays[0].size() - 1];
        for (int i = 1; i < arrays.size(); i++) {
            n = arrays[i].size();
            res = std::max(res, std::max(std::abs(arrays[i][n - 1] - min_val), 
                                         std::abs(max_val - arrays[i][0])));
            min_val = std::min(min_val, arrays[i][0]);
            max_val = std::max(max_val, arrays[i][n - 1]);
        }
        return res;
    }
};

class Solution {
    /**
     * @param {number[][]} arrays
     * @return {number}
     */
    maxDistance(arrays) {
        let res = 0;
        let n = arrays[0].length;
        let min_val = arrays[0][0];
        let max_val = arrays[0][arrays[0].length - 1];
        for (let i = 1; i < arrays.length; i++) {
            n = arrays[i].length;
            res = Math.max(res, Math.max(Math.abs(arrays[i][n - 1] - min_val), 
                                         Math.abs(max_val - arrays[i][0])));
            min_val = Math.min(min_val, arrays[i][0]);
            max_val = Math.max(max_val, arrays[i][n - 1]);
        }
        return res;
    }
}

public class Solution {
    public int MaxDistance(IList<IList<int>> arrays) {
        int res = 0;
        int n = arrays[0].Count;
        int min_val = arrays[0][0];
        int max_val = arrays[0][arrays[0].Count - 1];

        for (int i = 1; i < arrays.Count; i++) {
            n = arrays[i].Count;
            res = Math.Max(res, Math.Max(Math.Abs(arrays[i][n - 1] - min_val), 
                Math.Abs(max_val - arrays[i][0])));
            min_val = Math.Min(min_val, arrays[i][0]);
            max_val = Math.Max(max_val, arrays[i][n - 1]);
        }

        return res;
    }
}

func maxDistance(arrays [][]int) int {
    res := 0
    n := len(arrays[0])
    min_val := arrays[0][0]
    max_val := arrays[0][n-1]
    for i := 1; i < len(arrays); i++ {
        n = len(arrays[i])
        res = max(res, max(abs(arrays[i][n-1]-min_val),
                           abs(max_val-arrays[i][0])))
        min_val = min(min_val, arrays[i][0])
        max_val = max(max_val, arrays[i][n-1])
    }
    return res
}

func abs(x int) int {
    if x < 0 {
        return -x
    }
    return x
}

func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}

func min(a, b int) int {
    if a < b {
        return a
    }
    return b
}

class Solution {
    fun maxDistance(arrays: List<List<Int>>): Int {
        var res = 0
        var minVal = arrays[0][0]
        var maxVal = arrays[0][arrays[0].size - 1]
        for (i in 1 until arrays.size) {
            val n = arrays[i].size
            res = maxOf(res, maxOf(
                kotlin.math.abs(arrays[i][n - 1] - minVal),
                kotlin.math.abs(maxVal - arrays[i][0])
            ))
            minVal = minOf(minVal, arrays[i][0])
            maxVal = maxOf(maxVal, arrays[i][n - 1])
        }
        return res
    }
}

class Solution {
    func maxDistance(_ arrays: [[Int]]) -> Int {
        var res = 0
        var minVal = arrays[0][0]
        var maxVal = arrays[0][arrays[0].count - 1]
        for i in 1..<arrays.count {
            let n = arrays[i].count
            res = max(res, max(
                abs(arrays[i][n - 1] - minVal),
                abs(maxVal - arrays[i][0])
            ))
            minVal = min(minVal, arrays[i][0])
            maxVal = max(maxVal, arrays[i][n - 1])
        }
        return res
    }
}

Time & Space Complexity

Time complexity: $O(n)$
Space complexity: $O(1)$ constant space used

Where $n$ is the number of arrays in $arrays$

Common Pitfalls

Selecting Both Elements from the Same Array

The problem requires elements to come from two different arrays. A common mistake is to find the global minimum and maximum across all elements without ensuring they belong to different arrays. For example, if one array contains both the smallest and largest values, using them would violate the constraint. The single scan solution handles this by comparing the current array's values against previously seen min/max values, guaranteeing they come from different arrays.

Forgetting to Use Absolute Difference

Some solutions compute the difference without taking the absolute value. While the optimal solution always involves max - min (which is positive), intermediate calculations or edge cases might produce negative values. Always use abs() or structure your comparisons to ensure you're computing |a - b| rather than just a - b.

Not Exploiting the Sorted Property of Each Array

Each individual array is sorted in ascending order. Failing to recognize this leads to unnecessarily iterating through all elements when only the first (minimum) and last (maximum) elements of each array matter. The brute force solution that checks every pair of elements is correct but inefficient; the optimized approaches leverage the sorted property to achieve linear time.