2541. 使数组中所有元素相等的最小操作数 II

2023/01/28 leetcode 共 923 字,约 3 分钟

title

problem link

给你两个整数数组 nums1 和 nums2 ,两个数组长度都是 n ,再给你一个整数 k 。你可以对数组 nums1 进行以下操作:

  • 选择两个下标 i 和 j ,将 nums1[i] 增加 k ,将 nums1[j] 减少 k 。换言之,nums1[i] = nums1[i] + k 且 nums1[j] = nums1[j] - k 。

如果对于所有满足 0 <= i < n 都有 num1[i] == nums2[i] ,那么我们称 nums1 等于 nums2 。

请你返回使 nums1 等于 nums2 的 最少 操作数。如果没办法让它们相等,请你返回 -1 。

示例 1:

输入:nums1 = [4,3,1,4], nums2 = [1,3,7,1], k = 3
输出:2
解释:我们可以通过 2 个操作将 nums1 变成 nums2 。
第 1 个操作:i = 2 ,j = 0 。操作后得到 nums1 = [1,3,4,4] 。
第 2 个操作:i = 2 ,j = 3 。操作后得到 nums1 = [1,3,7,1] 。
无法用更少操作使两个数组相等。

示例 2:

输入:nums1 = [3,8,5,2], nums2 = [2,4,1,6], k = 1
输出:-1
解释:无法使两个数组相等。

提示:

  • n == nums1.length == nums2.length
  • 2 <= n <= 105
  • 0 <= nums1[i], nums2[j] <= 109
  • 0 <= k <= 105

solution

任选两个元素,一个+k,一个-k,问最少操作次数,使这个数组和另一个数组相同。累计操作次数即可,注意k为0的情况


class Solution:
    def minOperations(self, nums1: List[int], nums2: List[int], k: int) -> int:
        op1, op2 = 0, 0
        d = []
        for a, b in zip(nums1, nums2):
            d.append(a - b)
        if sum(d) != 0: return -1
        for v in d:
            if k == 0:
                # k==0时特判,不能相除法了。
                if v != 0: return -1
                continue
            if abs(v) % k != 0: return -1
            if v > 0:
                op1 += v // k
            else:
                op2 += (-v) // k
        if op1 != op2:
            return -1
        return op1

Search

    Table of Contents