3132. 找出与数组相加的整数 II

写写题，手痒了，顺便练练kotlin，结果......

给你两个整数数组 nums1 和 nums2。

从 nums1 中移除两个元素，并且所有其他元素都与变量 x 所表示的整数相加。如果 x 为负数，则表现为元素值的减少。

执行上述操作后，nums1 和 nums2 相等 。当两个数组中包含相同的整数，并且这些整数出现的频次相同时，两个数组 相等 。

返回能够实现数组相等的 最小 整数 x 。

示例 1:

输入：nums1 = [4,20,16,12,8], nums2 = [14,18,10]

输出：-2

解释：

移除 nums1 中下标为 [0,4] 的两个元素，并且每个元素与 -2 相加后，nums1 变为 [18,14,10] ，与 nums2 相等。

示例 2:

输入：nums1 = [3,5,5,3], nums2 = [7,7]

输出：2

解释：

移除 nums1 中下标为 [0,3] 的两个元素，并且每个元素与 2 相加后，nums1 变为 [7,7] ，与 nums2 相等。

提示：

• 3 <= nums1.length <= 200
• nums2.length == nums1.length - 2
• 0 <= nums1[i], nums2[i] <= 1000
• 测试用例以这样的方式生成：存在一个整数 x，nums1 中的每个元素都与 x 相加后，再移除两个元素，nums1 可以与 nums2 相等。

题解：

这道题挺有意思，刚开始的想法是枚举要删除的两个数，nums1的和减去这两个数后再减去nums2的和，然后除以nums2.length可得到一个候选x，然后判断是否符合条件（起初以为直接取最小的x就行了，经过这样处理后不用判断符合条件，不过似乎不行，例如2+5==3+4）。

放弃挣扎了，看官网题解，结果没看懂，还是得灵神的题解。

灵神解法kotlin版

fun minimumAddedInteger(nums1: IntArray, nums2: IntArray): Int {
    nums1.sort()
    nums2.sort()
    for (i in 2 downTo 1) {
        val x = nums2[0] - nums1[i]
        var j = 0
        for (k in i until nums1.size) {
            if (nums2[j] == nums1[k] + x && ++j == nums2.size) { // 这里处理很妙
                return x
            }
        }
    }
    return nums2[0] - nums1[0]
}

另一种kotlin版，其实差不多，写法不一样

fun minimumAddedInteger(nums1: IntArray, nums2: IntArray): Int {
    val n = nums2.size
    nums1.sort()
    nums2.sort()
    fun check(x: Int): Boolean {
        var diff = 0
        var i = 0
        for (j in 0 until n) {
            while (i < n + 2 && nums1[i] + x != nums2[j]) {
                i++
                diff++
            }
            if (i == n + 2) {
                return false
            }
            i++
        }
        return diff <= 2
    }
    var result = Int.MAX_VALUE / 2
    for (i in 0 until 3) {
        val x = nums2[0] - nums1[i]
        if (check(x)) {
            result = Math.min(result, x)
        }
    }
    return result
}

参考：原题链接