Skip to content

Latest commit

 

History

History
99 lines (88 loc) · 2.99 KB

1818.绝对差值和.md

File metadata and controls

99 lines (88 loc) · 2.99 KB

1818. 绝对差值和

给你两个正整数数组 nums1 和 nums2 ,数组的长度都是 n 。

数组 nums1 和 nums2 的 绝对差值和 定义为所有 |nums1[i] - nums2[i]|(0 <= i < n)的 总和(下标从 0 开始)。

你可以选用 nums1 中的 任意一个 元素来替换 nums1 中的 至多 一个元素,以 最小化 绝对差值和。

在替换数组 nums1 中最多一个元素 之后 ,返回最小绝对差值和。因为答案可能很大,所以需要对 109 + 7 取余 后返回。

|x| 定义为:

  • 如果 x >= 0 ,值为 x ,或者
  • 如果 x <= 0 ,值为 -x

示例 1:

输入:nums1 = [1,7,5], nums2 = [2,3,5]
输出:3
解释:有两种可能的最优方案:
- 将第二个元素替换为第一个元素:[1,7,5] => [1,1,5] ,或者
- 将第二个元素替换为第三个元素:[1,7,5] => [1,5,5]
两种方案的绝对差值和都是 |1-2| + (|1-3| 或者 |5-3|) + |5-5| = 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
解释:将第一个元素替换为第二个元素:[1,10,4,4,2,7] => [10,10,4,4,2,7]
绝对差值和为 |10-9| + |10-3| + |4-5| + |4-1| + |2-7| + |7-4| = 20

提示:

  • n == nums1.length
  • n == nums2.length
  • 1 <= n <= 10^5
  • 1 <= nums1[i], nums2[i] <= 10^5

解法

题目要求最多替换nums1中一个数字之后,最小的绝对值之和count,很明显我们需要做的是:

把nums2中的每个数,在nums1中找和它最接近的数字,并计算替换后的sum是否小于初始count。

为了快速在数组中找到一个数字,我们很容易想到需要二分查找,所以我们需要复制一份nums1,并排序它,用于二分查找。

func minAbsoluteSumDiff(nums1 []int, nums2 []int) int {
    copy1 := make([]int, len(nums1))
    copy(copy1, nums1)
    sort.Ints(copy1)
    var count int 
    for i := range nums1 {
        count = (count + absoluteVal(nums1[i]-nums2[i])) 
    }
    res := count
    for i, n := range nums2 {
        newN := midSearch(copy1, n)
        // 找一个与nums2[i]最接近的数字,如果不是nums1[i],就计算是否能使结果更小
        if newN != nums1[i] {
            tmpRes := count - absoluteVal(nums1[i]-n) + absoluteVal(newN-n)
            if tmpRes < res {
                res = tmpRes
            }
        }
    }
    return res % 1000000007
}

// 二分查找,在arr中找到和n最接近的数
func midSearch(arr []int, n int) int {
    s, e := 0, len(arr)-1
    for s < e-1 {
        m := (s+e)/2
        if arr[m] == n {
            return n
        }
        if arr[m] > n {
            e = m
        } else {
            s = m
        }
    } 
    if absoluteVal(n-arr[s]) < absoluteVal(n-arr[e]) {
        return arr[s]
    }
    return arr[e]
}

// 返回绝对值
func absoluteVal(i int) int {
    if i > 0 {
        return i
    }
    return -i
}