Formatted question description: https://leetcode.ca/all/1818.html
1818. Minimum Absolute Sum Difference
Level
Medium
Description
You are given two positive integer arrays nums1
and nums2
, both of length n
.
The absolute sum difference of arrays nums1
and nums2
is defined as the sum of |nums1[i] - nums2[i]|
for each 0 <= i < n
(0-indexed).
You can replace at most one element of nums1
with any other element in nums1
to minimize the absolute sum difference.
Return the minimum absolute sum difference after replacing at most one element in the array nums1
. Since the answer may be large, return it modulo 10^9 + 7
.
|x|
is defined as:
x
ifx >= 0
, or-x
ifx < 0
.
Example 1:
Input: nums1 = [1,7,5], nums2 = [2,3,5]
Output: 3
Explanation: There are two possible optimal solutions:
- Replace the second element with the first: [1,7,5] => [1,1,5], or
- Replace the second element with the third: [1,7,5] => [1,5,5].
Both will yield an absolute sum difference of | 1-2 | + ( | 1-3 | or | 5-3 | ) + | 5-5 | = 3. |
Example 2:
Input: nums1 = [2,4,6,8,10], nums2 = [2,4,6,8,10]
Output: 0
Explanation: nums1 is equal to nums2 so no replacement is needed. This will result in an absolute sum difference of 0.
Example 3:
Input: nums1 = [1,10,4,4,2,7], nums2 = [9,3,5,1,7,4]
Output: 20
Explanation: Replace the first element with the second: [1,10,4,4,2,7] => [10,10,4,4,2,7].
This yields an absolute sum difference of | 10-9 | + | 10-3 | + | 4-5 | + | 4-1 | + | 2-7 | + | 7-4 | = 20 |
Constraints:
n == nums1.length
n == nums2.length
1 <= n <= 10^5
1 <= nums1[i], nums2[i] <= 10^5
Solution
Use a tree set to store the numbers in nums1
, and calculate the original absolute sum difference. Then for each 0 <= i < n
, replace nums1[i]
with the two closest values of nums2[i]
and calculate the new absolute sum difference. Maintain the minimum absolute sum difference during the process. Finally, return the minimum absolute sum difference.
class Solution {
public int minAbsoluteSumDiff(int[] nums1, int[] nums2) {
final int MODULO = 1000000007;
TreeSet<Integer> set = new TreeSet<Integer>();
long sum = 0;
int length = nums1.length;
for (int i = 0; i < length; i++) {
sum += (long) Math.abs(nums1[i] - nums2[i]);
set.add(nums1[i]);
}
long minSum = sum;
for (int i = 0; i < length; i++) {
int num2 = nums2[i];
Integer newNumFloor = set.floor(num2), newNumCeiling = set.ceiling(num2);
long curSum = sum - (long) Math.abs(nums1[i] - nums2[i]);
if (newNumFloor != null) {
long curSumFloor = curSum + (long) Math.abs(newNumFloor - nums2[i]);
minSum = Math.min(minSum, curSumFloor);
}
if (newNumCeiling != null) {
long curSumCeiling = curSum + (long) Math.abs(newNumCeiling - nums2[i]);
minSum = Math.min(minSum, curSumCeiling);
}
}
return (int) (minSum % MODULO);
}
}