1818.绝对差值和
题目描述:
给你两个正整数数组nums1和nums2数组的长度都是n。
数组nums1和nums2的绝对差值和定义为|nums1[i] - nums2[i]|(0<=i<n)的总和(下标从0开始)。
你可以选用nums1中的任意一个元素替换nums1中的至多一个元素,以最小化绝对差值和。
在替换数组nums1中最多一个元素后,返回最小绝对差值和。因为答案可能很大,所以需要对10^9+7取余后返回。
|x|的定义为:如果x>=0,值为x;如果x<=0,值为-x。
示例1:
输入:nums1 = [1, 7, 5], nums2 = [2, 3, 5]
输出:3
解释:有两种可能的最优方案:一种是将nums1的第二个元素替换为第一个元素;另一种是将nums1的第二个元素替换为第三个元素。两种方案的绝对差值和都是3。
示例2:
输入:nums1 = [2, 4, 6, 8, 10], nums2 = [2, 4, 6, 8, 10]
输出:0
解释:nums1与nums2相等。不需要替换元素。绝对差值和为0。
示例3:
输入:nums1 = [1, 10, 4, 4, 2, 7], nums2 = [9, 3, 5, 1, 7, 4]
输出:20
解释:将nums1的第一个元素替换为第二个元素,绝对差值和为|10-9|+|10-3|+|4-5|+|4-1|+|2-7|+|7-4|=20
思路:
观察图片可知天际线只在建筑的左右端点出现,可以将高度直接排序,也可以用最小堆记录每个建筑变化的地方(按x坐标排序)。保存的高度为负数时加入建筑,保存的高度为正数时删除建筑。如果加入的新建筑是最高的,天际线发现变化;如果删除建筑后,最大高度变化了,天际线发生变化。使用SortedList判断最高高度是否变化即可。
代码:
class Solution:
def minAbsoluteSumDiff(self, nums1: List[int], nums2: List[int]) -> int:
n = len(nums1)
sum = 0
for i in range(n):
sum += abs(nums1[i] - nums2[i])
if sum == 0:
return 0
# 调整nums1的值
sorted_nums = sorted(set(nums1))
res = sum
for i in range(n):
index = bisect.bisect_left(sorted_nums, nums2[i])
# 找到nums1中与nums2[i]最接近的两个元素
# nums1[index-1]<nums2[i]
# nums1[index]>=nums[i]
tmp = float('inf')
if index > 0:
tmp = min(tmp, abs(sorted_nums[index-1]-nums2[i]))
if index < len(sorted_nums):
tmp = min(tmp, abs(sorted_nums[index]-nums2[i]))
res = min(res, sum-abs(nums1[i]-nums2[i])+tmp)
return res % (10**9+7)